Hoe gebruik je de “torch.argmax()”-methode in PyTorch?

In PyTorch is de “ fakkel.argmax() De methode is een ingebouwde functie die indices van maximale waarden van een bepaalde tensor over een bepaalde dimensie retourneert. Gebruikers gebruiken deze functie wanneer ze met tensoren werken en de index van de maximale waarde langs de gegeven dimensie van een tensor willen vinden. Bovendien kan deze methode ook nuttig zijn voor classificatie waarbij gebruikers willen weten welke klasse de hoogste waarschijnlijkheid heeft.

Deze blog zal een voorbeeld zijn van de methode om de “torch.argmax()” methode in PyTorch te gebruiken.

Hoe gebruik je de “torch.argmax()”-methode in PyTorch?

De methode “torch.argmax()” neemt elke 1D- of 2D-tensor als invoer en retourneert een tensor die de indices/indexen bevat van de maximale waarden langs de gegeven dimensie.

De syntaxis van de “torch.argmax()”-methode wordt hieronder gegeven:

fakkel. argmax ( < invoer_tensor > )

Om deze methode in PyTorch te gebruiken, doorloopt u de volgende voorbeelden voor een beter begrip:

Voorbeeld 1: Gebruik de “torch.argmax()”-methode met 1D-tensor

In het eerste voorbeeld zullen we een 1D-tensor maken en daarbij de methode “torch.argmax()” gebruiken. Laten we de onderstaande stapsgewijze procedure volgen:

Stap 1: Importeer de PyTorch-bibliotheek

Importeer eerst de “ fakkel ”-bibliotheek om de “torch.argmax()”-methode te gebruiken:

importeren fakkel

Stap 2: Maak een 1D-tensor

Maak vervolgens een 1D-tensor en druk de elementen ervan af. Hier creëren we het volgende “ Tientallen1 ” tensor uit een lijst met behulp van de “ fakkel.tensor() ” functie:

Tientallen1 = fakkel. tensor ( [ 5 , 0 , - 8 , 1 , 9 , 7 ] )

afdrukken ( Tientallen1 )

Hierdoor is een 1D-tensor ontstaan, zoals hieronder te zien is:

Stap 3: Zoek indices met maximale waarde

Maak nu gebruik van de “ fakkel.argmax() '-functie om de index/indices van de maximale waarde in de ' Tientallen1 ” tensor:

T1_ind = fakkel. argmax ( Tientallen1 )

Stap 4: Index van maximale waarde afdrukken

Geef ten slotte de index van de maximale waarde weer in de invoertensor:

afdrukken ( 'Indices:' , T1_ind )

De onderstaande uitvoer toont de index van de maximale waarde in de “ Tientallen1 ”tensor, d.w.z. 4. Het betekent dat de hoogste waarde van de tensor zich op de 4e index bevindt, namelijk “ 9 ”:

Voorbeeld 2: Gebruik de “torch.argmax()”-methode met 2D-tensor

In het tweede voorbeeld zullen we een 2D-tensor maken en daarbij de methode “torch.argmax()” gebruiken. Laten we de gegeven stappen volgen:

Stap 1: Importeer de PyTorch-bibliotheek

Importeer eerst de “ fakkel ”-bibliotheek om de “torch.argmax()”-methode te gebruiken:

importeren fakkel

Stap 2: Maak een 2D-tensor

Gebruik dan de “ fakkel.tensor() ”-functie om een 2D-tensor te maken en de elementen ervan af te drukken. Hier creëren we het volgende “ Tientallen2 “2D-tensor:

Tientallen2 = fakkel. tensor ( [ [ 4 , 1 , - 7 ] , [ vijftien , 6 , 0 ] , [ - 7 , 9 , 2 ] ] )

afdrukken ( Tientallen2 )

Hierdoor is een 2D-tensor ontstaan, zoals hieronder te zien is:

Stap 3: Zoek indices met maximale waarde

Zoek nu de index van de maximale waarde in de “ Tientallen2 ”tensor door gebruik te maken van de “ fakkel.argmax() ” functie:

T2_ind = fakkel. argmax ( Tientallen2 )

Stap 4: Index van maximale waarde afdrukken

Geef ten slotte de index van de maximale waarde weer in de invoertensor:

afdrukken ( 'Indices:' , T2_ind )

Volgens de onderstaande uitvoer is de index van de maximale waarde in de “ Tientallen2 ”tensor is “3”. Het betekent dat de hoogste waarde van de tensor zich op de derde index bevindt, namelijk “ vijftien ”:

Stap 5: Zoek indices van maximale waarde langs kolommen

Bovendien kunnen gebruikers ook de indexen/indices van de maximale waarden langs elke kolom van een tensor vinden. We kunnen bijvoorbeeld de “ dimmen=0 ”-argument met de functie “torch.argmax()”. Het vindt de indexen van de maximale waarden langs kolommen in de ' Tientallen2 ”tensor en drukt vervolgens die indices af:

col_index = fakkel. argmax ( Tientallen2 , afm = 0 )

afdrukken ( 'Indices in kolommen:' , col_index )

De onderstaande uitvoer toont de indexen van de maximale waarden langs elke kolom van de tensor:

Stap 6: Zoek indices van maximale waarde langs rijen

Op dezelfde manier kunnen gebruikers ook de indexen/indices van de maximale waarden langs elke rij van een tensor vinden. Gebruik bijvoorbeeld de “ dimmen=1 ”-argument met de functie “torch.argmax()” om de indices van de maximale waarden langs rijen in de “Tens2”-tensor te vinden en vervolgens die indices af te drukken:

rij_index = fakkel. argmax ( Tientallen2 , afm = 1 )

afdrukken ( 'Indices in rijen:' , rij_index )

De indexen van de maximale waarde langs elke rij van een “Tens2”-tensor zijn hieronder te zien:

We hebben efficiënt de methode uitgelegd om de methode “torch.argmax()” in PyTorch te gebruiken.

Opmerking : Hier heeft u toegang tot ons Google Colab Notebook koppeling .

Conclusie

Om de methode “torch.argmax()” in PyTorch te gebruiken, importeert u eerst de “ fakkel ' bibliotheek. Maak vervolgens de gewenste 1D- of 2D-tensor en bekijk de elementen ervan. Gebruik vervolgens de “ fakkel.argmax() ”methode om de indexen/indices van de maximale waarden in de tensor te vinden/berekenen. Bovendien kunnen gebruikers ook de indexen van de maximale waarde langs elke rij of kolom in de tensor vinden met behulp van de “ afm argument. Geef ten slotte de index van de maximale waarde weer in de invoertensor. Deze blog heeft de methode geïllustreerd om de methode “torch.argmax()” in PyTorch te gebruiken.

Hoe gebruik je de “torch.argmax()”-methode in PyTorch?