PyTorch adalah perpustakaan pembelajaran mesin yang memungkinkan pengguna bekerja dengan tensor. Tensor adalah struktur data penting yang dapat memiliki berbagai jenis data dan bentuk. Terkadang, pengguna mungkin ingin melakukan operasi perluasan pada tensor untuk memperbesar ukurannya. Operasi perluasan memungkinkan pengguna mengulangi tensor sepanjang dimensi tertentu. PyTorch menyediakan atribut “expand()” yang menggunakan tensor dan daftar ukuran sebagai input. Ia mengembalikan tensor baru yang memiliki data yang sama tetapi dengan dimensi berbeda.
Artikel ini akan mengilustrasikan metode penggunaan operasi perluasan pada tensor di PyTorch.
Bagaimana Cara Menggunakan Operasi Perluasan di PyTorch?
Untuk menggunakan operasi perluasan di PyTorch, lihat langkah-langkah yang disediakan di bawah ini:
- Impor perpustakaan PyTorch
- Buat tensor yang diinginkan
- Lihat ukuran tensor masukan
- Perluas tensor menggunakan “ memperluas() ” atribut
- Tampilkan tensor yang diperluas dan ukurannya
Langkah 1: Impor Perpustakaan PyTorch
Pertama, impor “ obor ” perpustakaan untuk menggunakan operasi perluasan:
impor obor
Langkah 2: Buat Tensor
Kemudian, buat tensor yang diinginkan menggunakan “ obor.tensor() ” berfungsi dan mencetak elemen-elemennya. Di sini, kami membuat yang berikut ini “ Puluhan ”tensor:
Puluhan = obor. tensor ( [ [ 2 ] , [ 4 ] , [ 6 ] ] )
mencetak ( Puluhan )
Pada output di bawah ini, tensor telah berhasil dibuat:
Langkah 3: Lihat Ukuran Tensor Input
Selanjutnya, lihat ukuran “ Puluhan ” tensor menggunakan “ ukuran() ” atribut:
mencetak ( 'Ukuran Tensor:' , Puluhan. ukuran ( ) )Berdasarkan keluaran di bawah ini, ukuran tensornya adalah 3×1:
Langkah 4: Perluas Tensor
Sekarang, gunakan “ memperluas() ” atribut untuk melakukan operasi perluasan dan memperluas tensor ke dimensi baru. Diperlukan untuk memberikan tensor dan daftar ukuran sebagai masukan. Di sini, kami memperluas tensor ke ukuran 3×4:
Exp_tens = Puluhan. memperluas ( 3 , 4 )Langkah 5: Tampilkan Tensor yang Diperluas dan Ukurannya
Terakhir, cetak elemen tensor yang diperluas dan ukurannya:
mencetak ( Exp_tens )mencetak ( Exp_tens. ukuran ( ) )
Output di bawah ini menunjukkan tensor yang diperluas dan ukurannya, yaitu 3×4. Hal ini menunjukkan bahwa operasi yang diperluas telah berhasil dilakukan:
Kami telah menjelaskan metode efisien untuk menggunakan operasi perluasan di PyTorch
Catatan : Anda dapat mengakses Notebook Google Colab kami di sini tautan .
Kesimpulan
Untuk menggunakan operasi perluasan di PyTorch, pertama-tama impor pustaka obor. Kemudian, buat tensor yang diinginkan dan lihat elemen serta ukurannya. Selanjutnya, gunakan “ memperluas() ” atribut untuk memperluas tensor masukan. Terakhir, cetak tensor yang diperluas dan lihat ukurannya. Artikel ini telah mendemonstrasikan metode untuk menggunakan operasi perluasan pada tensor di PyTorch.