Kuinka käyttää 'torch.argmax()' -menetelmää PyTorchissa?

PyTorchissa ' torch.argmax() ”-menetelmä on sisäänrakennettu funktio, joka palauttaa tietyn tensorin maksimiarvojen indeksit tietyn ulottuvuuden yli. Käyttäjät käyttävät tätä toimintoa, kun he työskentelevät tensoreiden kanssa ja haluavat löytää maksimiarvon indeksin tensorin annetulla ulottuvuudella. Lisäksi tämä menetelmä voi olla hyödyllinen myös luokittelussa, jossa käyttäjät haluavat tietää, millä luokalla on suurin todennäköisyys.

Tämä blogi on esimerkki menetelmästä, jolla PyTorchissa käytetään 'torch.argmax()' -menetelmää.

Kuinka käyttää 'torch.argmax()' -menetelmää PyTorchissa?

Menetelmä 'torch.argmax()' ottaa syötteeksi minkä tahansa 1D- tai 2D-tensorin ja palauttaa tensorin, joka sisältää maksimiarvojen indeksit/indeksit annetulla ulottuvuudella.

'Torch.argmax()'-menetelmän syntaksi on annettu alla:

taskulamppu. argmax ( < input_tensor > )

Jos haluat käyttää tätä menetelmää PyTorchissa, käy läpi seuraavat esimerkit ymmärtääksesi paremmin:

Esimerkki 1: Käytä 'torch.argmax()' -menetelmää 1D-tensorin kanssa

Ensimmäisessä esimerkissä luomme 1D-tensorin ja käytämme sen kanssa menetelmää 'torch.argmax()'. Noudatetaan alla olevaa vaiheittaista menettelyä:

Vaihe 1: Tuo PyTorch-kirjasto

Tuo ensin ' taskulamppu '-kirjasto käyttää 'torch.argmax()'-menetelmää:

tuonti taskulamppu

Vaihe 2: Luo 1D-tensori

Luo sitten 1D-tensori ja tulosta sen elementit. Täällä luomme seuraavan ' Kymmeniä 1 ' tensoroi luettelosta käyttämällä ' torch.tensor() ”toiminto:

Kymmeniä 1 = taskulamppu. tensori ( [ 5 , 0 , - 8 , 1 , 9 , 7 ] )

Tulosta ( Kymmeniä 1 )

Tämä on luonut 1D-tensorin, kuten alla näkyy:

Vaihe 3: Etsi enimmäisarvon indeksit

Käytä nyt ' torch.argmax() ' -toiminto löytääksesi enimmäisarvon indeksin/indeksit ' Kymmeniä 1 'tensori:

T1_ind = taskulamppu. argmax ( Kymmeniä 1 )

Vaihe 4: Tulosta enimmäisarvon indeksi

Näytä lopuksi maksimiarvon indeksi syöttötensorissa:

Tulosta ( 'Indeksit:' , T1_ind )

Alla oleva tulos näyttää maksimiarvon indeksin ' Kymmeniä 1 ' tensori eli 4. Se tarkoittaa, että tensorin suurin arvo on 4. indeksissä, joka on ' 9 ':

Esimerkki 2: Käytä 'torch.argmax()' -menetelmää 2D-tensorin kanssa

Toisessa esimerkissä luomme 2D-tensorin ja käytämme sen kanssa menetelmää 'torch.argmax()'. Noudatetaan annettuja vaiheita:

Vaihe 1: Tuo PyTorch-kirjasto

Tuo ensin ' taskulamppu '-kirjasto käyttää 'torch.argmax()'-menetelmää:

tuonti taskulamppu

Vaihe 2: Luo 2D-tensori

Käytä sitten ' torch.tensor() ”-toiminnolla voit luoda 2D-tensorin ja tulostaa sen elementit. Täällä luomme seuraavan ' Kymmeniä 2 '2D-tensori:

Kymmeniä 2 = taskulamppu. tensori ( [ [ 4 , 1 , - 7 ] , [ viisitoista , 6 , 0 ] , [ - 7 , 9 , 2 ] ] )

Tulosta ( Kymmeniä 2 )

Tämä on luonut 2D-tensorin, kuten alla näkyy:

Vaihe 3: Etsi enimmäisarvon indeksit

Etsi nyt maksimiarvon indeksi ' Kymmeniä 2 ' tensori käyttämällä ' torch.argmax() ”toiminto:

T2_ind = taskulamppu. argmax ( Kymmeniä 2 )

Vaihe 4: Tulosta enimmäisarvon indeksi

Lopuksi näytä maksimiarvon indeksi syöttötensorissa:

Tulosta ( 'Indeksit:' , T2_ind )

Alla olevan lähdön mukaan maksimiarvon indeksi ' Kymmeniä 2 ' tensori on '3'. Se tarkoittaa, että tensorin suurin arvo on 3. indeksissä, joka on ' viisitoista ':

Vaihe 5: Etsi enimmäisarvon indeksit sarakkeiden vierestä

Lisäksi käyttäjät voivat myös löytää maksimiarvojen indeksit/indeksit jokaisesta tensorin sarakkeesta. Voimme käyttää esimerkiksi ' himmeä = 0 ' argumentti 'torch.argmax()'-funktiolla. Se löytää maksimiarvojen indeksit sarakkeista ' Kymmeniä 2 ”tensoroi ja tulostaa sitten nämä indeksit:

col_index = taskulamppu. argmax ( Kymmeniä 2 , himmeä = 0 )

Tulosta ( 'Indeksit sarakkeissa:' , col_index )

Alla oleva tulos näyttää maksimiarvojen indeksit kussakin tensorin sarakkeessa:

Vaihe 6: Etsi enimmäisarvon indeksit rivien varrelta

Vastaavasti käyttäjät voivat myös löytää maksimiarvojen indeksit/indeksit tensorin jokaiselta riviltä. Käytä esimerkiksi ' himmeä = 1 ' argumentti 'torch.argmax()'-funktiolla löytääksesi maksimiarvojen indeksit 'Tens2'-tensorin riveiltä ja tulostaa sitten nämä indeksit:

rivi_indeksi = taskulamppu. argmax ( Kymmeniä 2 , himmeä = 1 )

Tulosta ( 'Indeksit riveissä:' , rivi_indeksi )

Maksimiarvon indeksit 'Tens2'-tensorin jokaisella rivillä näkyvät alla:

Olemme selittäneet tehokkaasti menetelmän käyttää 'torch.argmax()'-menetelmää PyTorchissa.

Huomautus : Pääset käyttämään Google Colab -muistikirjaamme tästä linkki .

Johtopäätös

Jos haluat käyttää 'torch.argmax()'-menetelmää PyTorchissa, tuo ensin ' taskulamppu ' kirjasto. Luo sitten haluamasi 1D- tai 2D-tensori ja tarkastele sen elementtejä. Käytä seuraavaksi ' torch.argmax() ” menetelmä löytää/laskea tensorin maksimiarvojen indeksit/indeksit. Lisäksi käyttäjät voivat löytää maksimiarvon indeksit jokaiselta tensorin riviltä tai sarakkeelta käyttämällä ' himmeä ' Perustelu. Lopuksi näytä maksimiarvon indeksi syöttötensorissa. Tämä blogi on esimerkki menetelmästä käyttää 'torch.argmax()'-menetelmää PyTorchissa.

Kuinka käyttää 'torch.argmax()' -menetelmää PyTorchissa?