PyTorch ialah perpustakaan pembelajaran mesin yang membenarkan pengguna bekerja dengan tensor. Tensor ialah struktur data penting yang boleh mempunyai jenis data dan bentuk yang berbeza. Kadangkala, pengguna mungkin mahu melakukan operasi kembangan pada tensor untuk memanjangkan saiznya. Operasi kembangkan membolehkan pengguna mengulang tensor sepanjang dimensi tertentu. PyTorch menyediakan atribut 'expand()' yang mengambil tensor dan senarai saiz sebagai input. Ia mengembalikan tensor baharu yang mempunyai data yang sama tetapi dengan dimensi yang berbeza.
Artikel ini akan menggambarkan kaedah untuk menggunakan operasi kembangkan pada tensor dalam PyTorch.
Bagaimana untuk Menggunakan Operasi Kembangkan dalam PyTorch?
Untuk menggunakan operasi kembangkan dalam PyTorch, lihat langkah-langkah yang disediakan di bawah:
- Import perpustakaan PyTorch
- Buat tensor yang dikehendaki
- Lihat saiz tensor input
- Kembangkan tensor menggunakan “ kembangkan() ” atribut
- Paparkan tensor yang diperluas dan saiznya
Langkah 1: Import Pustaka PyTorch
Pertama, import ' obor ” perpustakaan untuk menggunakan operasi kembangkan:
import obor
Langkah 2: Buat Tensor
Kemudian, buat tensor yang dikehendaki menggunakan ' torch.tensor() ” berfungsi dan mencetak elemennya. Di sini, kami mencipta yang berikut ' Berpuluh-puluh ” tensor:
Berpuluh-puluh = obor. tensor ( [ [ 2 ] , [ 4 ] , [ 6 ] ] )
cetak ( Berpuluh-puluh )
Dalam output di bawah, tensor telah berjaya dibuat:
Langkah 3: Lihat Saiz Tensor Input
Seterusnya, lihat saiz ' yang dibuat di atas Berpuluh-puluh ” tensor menggunakan “ saiz() ” atribut:
cetak ( 'Saiz Tensor:' , Berpuluh-puluh. saiz ( ) )Mengikut output di bawah, saiz tensor ialah 3×1:
Langkah 4: Kembangkan Tensor
Sekarang, gunakan ' kembangkan() ” atribut untuk melaksanakan operasi kembangkan dan memanjangkan tensor ke dimensi baharu. Ia dikehendaki menyediakan tensor dan senarai saiz sebagai input. Di sini, kami mengembangkan tensor kepada saiz 3×4:
Exp_tens = Berpuluh-puluh. mengembang ( 3 , 4 )Langkah 5: Paparkan Tensor Dikembangkan dan Saiznya
Akhir sekali, cetak elemen tensor yang diperluas dan saiznya:
cetak ( Exp_tens )cetak ( Exp_tens. saiz ( ) )
Output di bawah menunjukkan tensor yang diperluas dan saiznya iaitu, 3×4. Ini menunjukkan bahawa operasi yang diperluaskan telah berjaya dilakukan:
Kami telah menerangkan kaedah yang cekap untuk menggunakan operasi kembangkan dalam PyTorch
Catatan : Anda boleh mengakses Buku Nota Google Colab kami di sini pautan .
Kesimpulan
Untuk menggunakan operasi kembangkan dalam PyTorch, pertama, import pustaka obor. Kemudian, cipta tensor yang dikehendaki dan lihat elemen dan saiznya. Seterusnya, gunakan ' kembangkan() ” atribut untuk mengembangkan tensor input. Akhir sekali, cetak tensor yang dikembangkan dan lihat saiznya. Artikel ini telah menunjukkan kaedah untuk menggunakan operasi kembangkan pada tensor dalam PyTorch.