Bagaimana Cara Menggunakan Metode “torch.argmax()” di PyTorch?

Di PyTorch, “ obor.argmax() Metode ” adalah fungsi bawaan yang mengembalikan indeks nilai maksimum tensor tertentu pada dimensi tertentu. Pengguna menggunakan fungsi ini saat mereka bekerja dengan tensor dan ingin mencari indeks nilai maksimum sepanjang dimensi tensor tertentu. Selain itu, metode ini juga berguna untuk klasifikasi dimana pengguna ingin mengetahui kelas mana yang memiliki probabilitas tertinggi.

Blog ini akan memberikan contoh metode penggunaan metode “torch.argmax()” di PyTorch.

Bagaimana Cara Menggunakan Metode “torch.argmax()” di PyTorch?

Metode “torch.argmax()” menggunakan tensor 1D atau 2D apa pun sebagai masukan dan mengembalikan tensor yang berisi indeks/indeks nilai maksimum sepanjang dimensi tertentu.

Sintaks metode “torch.argmax()” diberikan di bawah ini:

obor. argmax ( < masukan_tensor > )

Untuk menggunakan metode ini di PyTorch, lihat contoh berikut untuk pemahaman yang lebih baik:

Contoh 1: Gunakan Metode “torch.argmax()” Dengan Tensor 1D

Pada contoh pertama, kita akan membuat tensor 1D dan menggunakan metode “torch.argmax()” dengannya. Mari ikuti prosedur langkah demi langkah di bawah ini:

Langkah 1: Impor Perpustakaan PyTorch

Pertama, impor “ obor ” perpustakaan untuk menggunakan metode “torch.argmax()”:

impor obor

Langkah 2: Buat Tensor 1D

Kemudian, buat tensor 1D dan cetak elemennya. Di sini, kami membuat yang berikut ini “ Puluhan1 ” Tensor dari daftar menggunakan “ obor.tensor() ' fungsi:

Puluhan1 = obor. tensor ( [ 5 , 0 , - 8 , 1 , 9 , 7 ] )

mencetak ( Puluhan1 )

Ini telah menciptakan tensor 1D seperti yang terlihat di bawah ini:

Langkah 3: Temukan Indeks Nilai Maksimum

Sekarang, manfaatkan “ obor.argmax() ” berfungsi untuk mencari indeks/indeks nilai maksimal pada kolom “ Puluhan1 ”tensor:

T1_ind = obor. argmax ( Puluhan1 )

Langkah 4: Cetak Indeks Nilai Maksimum

Terakhir, tampilkan indeks nilai maksimum pada tensor masukan:

mencetak ( 'Indeks:' , T1_ind )

Output di bawah ini menunjukkan indeks nilai maksimum di “ Puluhan1 ” tensor yaitu 4. Artinya nilai tensor tertinggi ada pada indeks ke-4 yaitu “ 9 ”:

Contoh 2: Gunakan Metode “torch.argmax()” Dengan Tensor 2D

Pada contoh kedua, kita akan membuat tensor 2D dan menggunakan metode “torch.argmax()” dengannya. Mari ikuti langkah-langkah yang disediakan:

Langkah 1: Impor Perpustakaan PyTorch

Pertama, impor “ obor ” perpustakaan untuk menggunakan metode “torch.argmax()”:

impor obor

Langkah 2: Buat Tensor 2D

Kemudian, gunakan “ obor.tensor() ” berfungsi untuk membuat tensor 2D dan mencetak elemen-elemennya. Di sini, kami membuat yang berikut ini “ Puluhan2 “tensor 2D:

Puluhan2 = obor. tensor ( [ [ 4 , 1 , - 7 ] , [ limabelas , 6 , 0 ] , [ - 7 , 9 , 2 ] ] )

mencetak ( Puluhan2 )

Ini telah menciptakan tensor 2D seperti yang terlihat di bawah ini:

Langkah 3: Temukan Indeks Nilai Maksimum

Sekarang, cari indeks nilai maksimum di kolom “ Puluhan2 ” Tensor dengan memanfaatkan “ obor.argmax() ' fungsi:

T2_ind = obor. argmax ( Puluhan2 )

Langkah 4: Cetak Indeks Nilai Maksimum

Terakhir, tampilkan indeks nilai maksimum pada tensor masukan:

mencetak ( 'Indeks:' , T2_ind )

Menurut keluaran di bawah ini, indeks nilai maksimum di “ Puluhan2 ”tensornya adalah “3”. Artinya nilai tensor tertinggi ada pada indeks ke-3 yaitu “ limabelas ”:

Langkah 5: Temukan Indeks Nilai Maksimum Sepanjang Kolom

Selain itu, pengguna juga dapat menemukan indeks/indeks nilai maksimum di setiap kolom tensor. Misalnya, kita dapat menggunakan “ redup=0 argumen ” dengan fungsi “torch.argmax()”. Ia menemukan indeks nilai maksimum di sepanjang kolom di “ Puluhan2 ” tensor lalu mencetak indeks tersebut:

col_index = obor. argmax ( Puluhan2 , redup = 0 )

mencetak ( 'Indeks dalam kolom:' , col_index )

Output di bawah ini menunjukkan indeks nilai maksimum di setiap kolom tensor:

Langkah 6: Temukan Indeks Nilai Maksimum Sepanjang Baris

Demikian pula, pengguna juga dapat menemukan indeks/indeks nilai maksimum di sepanjang setiap baris tensor. Misalnya, gunakan “ redup=1 ” argumen dengan fungsi “torch.argmax()” untuk menemukan indeks nilai maksimum di sepanjang baris dalam tensor “Tens2” dan kemudian mencetak indeks tersebut:

baris_indeks = obor. argmax ( Puluhan2 , redup = 1 )

mencetak ( 'Indeks dalam baris:' , baris_indeks )

Indeks nilai maksimum pada setiap baris tensor “Tens2” dapat dilihat di bawah:

Kami telah menjelaskan secara efisien metode untuk menggunakan metode “torch.argmax()” di PyTorch.

Catatan : Anda dapat mengakses Notebook Google Colab kami di sini tautan .

Kesimpulan

Untuk menggunakan metode “torch.argmax()” di PyTorch, pertama-tama, impor “ obor ' perpustakaan. Kemudian, buat tensor 1D atau 2D yang diinginkan dan lihat elemennya. Selanjutnya, gunakan “ obor.argmax() ” metode untuk mencari/menghitung indeks/indeks nilai maksimum pada tensor. Selain itu, pengguna juga dapat menemukan indeks nilai maksimum di sepanjang setiap baris atau kolom di tensor menggunakan “ redup argumen. Terakhir, tampilkan indeks nilai maksimum pada tensor masukan. Blog ini telah mencontohkan metode penggunaan metode “torch.argmax()” di PyTorch.

Bagaimana Cara Menggunakan Metode “torch.argmax()” di PyTorch?