Bu blog, PyTorch'ta “torch.argmax()” yöntemini kullanma yöntemini örneklendirecektir.
PyTorch'ta “torch.argmax()” Yöntemi Nasıl Kullanılır?
'torch.argmax()' yöntemi herhangi bir 1D veya 2D tensörü girdi olarak alır ve verilen boyut boyunca maksimum değerlerin indekslerini/indekslerini içeren bir tensör döndürür.
“torch.argmax()” metodunun sözdizimi aşağıda verilmiştir:
meşale. argmaks ( < giriş_tensörü > )
Bu yöntemi PyTorch'ta kullanmak için daha iyi anlamak amacıyla aşağıdaki örnekleri inceleyin:
Örnek 1: 1D Tensörle “torch.argmax()” Yöntemini Kullanma
İlk örnekte 1 boyutlu bir tensör oluşturup onunla “torch.argmax()” yöntemini kullanacağız. Aşağıdaki adım adım prosedürü takip edelim:
1. Adım: PyTorch Kitaplığını İçe Aktarın
İlk önce “ meşale “torch.argmax()” yöntemini kullanmak için ” kütüphanesi:
içe aktarmak meşaleAdım 2: 1D Tensör Oluşturun
Ardından bir 1 boyutlu tensör oluşturun ve elemanlarını yazdırın. Burada aşağıdakileri oluşturuyoruz” Onlarca1 “ kullanarak bir listeden tensör meşale.tensor() ' işlev:
Onlarca1 = meşale. tensör ( [ 5 , 0 , - 8 , 1 , 9 , 7 ] )
Yazdır ( Onlarca1 )
Bu, aşağıda görüldüğü gibi bir 1 boyutlu tensör yarattı:
Adım 3: Maksimum Değer Endekslerini Bulun
Şimdi “ meşale.argmax() “ içindeki maksimum değerin indeksini/endekslerini bulmak için ” fonksiyonu Onlarca1 ” tensör:
T1_ind = meşale. argmaks ( Onlarca1 )Adım 4: Maksimum Değer Dizinini Yazdırın
Son olarak, giriş tensöründe maksimum değerin indeksini görüntüleyin:
Yazdır ( 'Endeksler:' , T1_ind )Aşağıdaki çıktı “” içindeki maksimum değerin indeksini gösterir. Onlarca1 ” tensör yani 4. Tensörün en yüksek değerinin 4. indekste olduğu anlamına gelir, yani “ 9 ”:
Örnek 2: 2D Tensörle “torch.argmax()” Yöntemini Kullanma
İkinci örnekte 2 boyutlu bir tensör oluşturup onunla “torch.argmax()” yöntemini kullanacağız. Sağlanan adımları takip edelim:
1. Adım: PyTorch Kitaplığını İçe Aktarın
İlk önce “ meşale “torch.argmax()” yöntemini kullanmak için ” kütüphanesi:
içe aktarmak meşaleAdım 2: 2D Tensör Oluşturun
Daha sonra “ meşale.tensor() 2 boyutlu bir tensör oluşturmak ve elemanlarını yazdırmak için ” işlevi. Burada aşağıdakileri oluşturuyoruz” Onlarca2 “2 boyutlu tensör:
Onlarca2 = meşale. tensör ( [ [ 4 , 1 , - 7 ] , [ on beş , 6 , 0 ] , [ - 7 , 9 , 2 ] ] )Yazdır ( Onlarca2 )
Bu, aşağıda görüldüğü gibi bir 2 boyutlu tensör yarattı:
Adım 3: Maksimum Değer Endekslerini Bulun
Şimdi “” içindeki maksimum değerin indeksini bulun. Onlarca2 “ tensörünü kullanarak meşale.argmax() ' işlev:
T2_ind = meşale. argmaks ( Onlarca2 )Adım 4: Maksimum Değer Dizinini Yazdırın
Son olarak, giriş tensöründe maksimum değerin indeksini görüntüleyin:
Yazdır ( 'Endeksler:' , T2_ind )Aşağıdaki çıktıya göre “” içindeki maksimum değerin indeksi Onlarca2 tensörü “3”tür. Bu, tensörün en yüksek değerinin 3. indekste olduğu anlamına gelir; on beş ”:
Adım 5: Sütunlar Boyunca Maksimum Değerin İndekslerini Bulun
Ayrıca kullanıcılar, bir tensörün her sütunu boyunca maksimum değerlerin indekslerini/indekslerini de bulabilirler. Örneğin “” ifadesini kullanabiliriz. loş=0 “torch.argmax()” işleviyle ” argümanı. Maksimum değerlerin indekslerini “ sütunları boyunca bulur. Onlarca2 ” tensörünü kullanır ve ardından bu endeksleri yazdırır:
sütun_index'i = meşale. argmaks ( Onlarca2 , loş = 0 )Yazdır ( 'Sütunlardaki dizinler:' , sütun_index'i )
Aşağıdaki çıktı, tensörün her bir sütunu boyunca maksimum değerlerin indekslerini gösterir:
Adım 6: Satırlar Boyunca Maksimum Değerin Endekslerini Bulun
Benzer şekilde kullanıcılar, bir tensörün her satırı boyunca maksimum değerlerin indekslerini/indekslerini de bulabilirler. Örneğin, “ loş=1 'Tens2' tensöründeki satırlar boyunca maksimum değerlerin endekslerini bulmak ve ardından bu endeksleri yazdırmak için 'torch.argmax()' işleviyle 'bağımsız değişkeni:
satır_index'i = meşale. argmaks ( Onlarca2 , loş = 1 )Yazdır ( 'Satırlardaki endeksler:' , satır_index'i )
Bir “Tens2” tensörünün her satırındaki maksimum değerin indeksleri aşağıda görülebilir:
PyTorch'ta “torch.argmax()” metodunu kullanma yöntemini verimli bir şekilde anlattık.
Not : Google Colab Not Defterimize buradan erişebilirsiniz bağlantı .
Çözüm
PyTorch'ta “torch.argmax()” yöntemini kullanmak için öncelikle “ meşale ' kütüphane. Ardından istediğiniz 1B veya 2B tensörü oluşturun ve öğelerini görüntüleyin. Daha sonra “ meşale.argmax() Tensördeki maksimum değerlerin indekslerini/indekslerini bulma/hesaplama yöntemi. Ayrıca kullanıcılar tensördeki her satır veya sütun boyunca maksimum değerin indekslerini “ loş ' argüman. Son olarak, giriş tensöründe maksimum değerin indeksini görüntüleyin. Bu blog, PyTorch'ta “torch.argmax()” yöntemini kullanma yöntemini örneklemiştir.