Hvordan bruke 'torch.argmax()'-metoden i PyTorch?

I PyTorch er ' torch.argmax() ”-metoden er en innebygd funksjon som returnerer indekser av maksimalverdier for en bestemt tensor over en gitt dimensjon. Brukere bruker denne funksjonen når de jobber med tensorer og ønsker å finne indeksen for maksimumsverdien langs en tensors gitte dimensjon. Dessuten kan denne metoden også være nyttig for klassifisering der brukere ønsker å vite hvilken klasse som har høyest sannsynlighet.

Denne bloggen vil eksemplifisere metoden for å bruke 'torch.argmax()'-metoden i PyTorch.

Hvordan bruke 'torch.argmax()'-metoden i PyTorch?

'torch.argmax()'-metoden tar en hvilken som helst 1D- eller 2D-tensor som input og returnerer en tensor som inneholder indeksene/indeksene til maksimalverdiene langs den gitte dimensjonen.

Syntaksen til 'torch.argmax()'-metoden er gitt nedenfor:

lommelykt. argmax ( < input_tensor > )

For å bruke denne metoden i PyTorch, gå gjennom følgende eksempler for en bedre forståelse:

Eksempel 1: Bruk «torch.argmax()»-metoden med 1D-tensor

I det første eksemplet vil vi lage en 1D-tensor og bruke 'torch.argmax()'-metoden med den. La oss følge trinn-for-trinn prosedyren nedenfor:

Trinn 1: Importer PyTorch-biblioteket

Først importerer du ' lommelykt '-biblioteket for å bruke 'torch.argmax()'-metoden:

import lommelykt

Trinn 2: Lag 1D Tensor

Lag deretter en 1D-tensor og skriv ut elementene. Her lager vi følgende ' Tiere 1 ' tensor fra en liste med ' torch.tensor() ' funksjon:

Tiere 1 = lommelykt. tensor ( [ 5 , 0 , - 8 , 1 , 9 , 7 ] )

skrive ut ( Tiere 1 )

Dette har skapt en 1D-tensor som vist nedenfor:

Trinn 3: Finn indekser for maksimal verdi

Bruk nå ' torch.argmax() '-funksjonen for å finne indeksen/indeksene for maksimalverdien i ' Tiere 1 ' tensor:

T1_ind = lommelykt. argmax ( Tiere 1 )

Trinn 4: Skriv ut indeks med maksimal verdi

Til slutt, vis maksimumsverdiens indeks i inngangstensoren:

skrive ut ( 'Indekser:' , T1_ind )

Utgangen nedenfor viser indeksen for maksimalverdien i ' Tiere 1 ' tensor dvs. 4. Det betyr at den høyeste verdien av tensoren er ved den fjerde indeksen som er ' 9 ':

Eksempel 2: Bruk «torch.argmax()»-metoden med 2D-tensor

I det andre eksemplet vil vi lage en 2D-tensor og bruke 'torch.argmax()'-metoden med den. La oss følge de angitte trinnene:

Trinn 1: Importer PyTorch-biblioteket

Først importerer du ' lommelykt '-biblioteket for å bruke 'torch.argmax()'-metoden:

import lommelykt

Trinn 2: Lag 2D-tensor

Deretter bruker du ' torch.tensor() ”-funksjon for å lage en 2D-tensor og skrive ut elementene. Her lager vi følgende ' Tiere 2 '2D tensor:

Tiere 2 = lommelykt. tensor ( [ [ 4 , 1 , - 7 ] , [ femten , 6 , 0 ] , [ - 7 , 9 , 2 ] ] )

skrive ut ( Tiere 2 )

Dette har skapt en 2D-tensor som vist nedenfor:

Trinn 3: Finn indekser for maksimal verdi

Finn nå indeksen for maksimalverdien i ' Tiere 2 ' tensor ved å bruke ' torch.argmax() ' funksjon:

T2_ind = lommelykt. argmax ( Tiere 2 )

Trinn 4: Skriv ut indeks med maksimal verdi

Til slutt, vis maksimalverdiens indeks i inngangstensoren:

skrive ut ( 'Indekser:' , T2_ind )

I henhold til utgangen nedenfor, er indeksen for maksimalverdien i ' Tiere 2 ' tensor er '3'. Det betyr at den høyeste verdien av tensoren er ved den tredje indeksen som er ' femten ':

Trinn 5: Finn indekser for maksimal verdi langs kolonner

Dessuten kan brukere også finne indeksene/indeksene for maksimalverdiene langs hver kolonne i en tensor. For eksempel kan vi bruke ' dim=0 ” argument med funksjonen “torch.argmax()”. Den finner maksimumsverdienes indekser langs kolonner i ' Tiere 2 ' tensor og skriver deretter ut disse indeksene:

col_index = lommelykt. argmax ( Tiere 2 , dempet = 0 )

skrive ut ( 'Indekser i kolonner:' , col_index )

Utgangen nedenfor viser indeksene til maksimalverdiene langs hver kolonne i tensoren:

Trinn 6: Finn indekser for maksimal verdi langs rader

På samme måte kan brukere også finne indeksene/indeksene for maksimalverdiene langs hver rad i en tensor. Bruk for eksempel ' dim=1 ”-argumentet med “torch.argmax()”-funksjonen for å finne maksimumsverdienes indekser langs rader i “Tens2”-tensoren og deretter skrive ut disse indeksene:

rad_indeks = lommelykt. argmax ( Tiere 2 , dempet = 1 )

skrive ut ( 'Indekser i rader:' , rad_indeks )

Maksimalverdiens indekser langs hver rad av en 'Tens2' tensor kan sees nedenfor:

Vi har effektivt forklart metoden for å bruke 'torch.argmax()'-metoden i PyTorch.

Merk : Du kan få tilgang til Google Colab Notebook på denne link .

Konklusjon

For å bruke «torch.argmax()»-metoden i PyTorch, importer først « lommelykt ' bibliotek. Lag deretter ønsket 1D- eller 2D-tensor og se elementene. Deretter bruker du ' torch.argmax() ” metode for å finne/beregne indeksene/indeksene til maksimalverdiene i tensoren. Dessuten kan brukere også finne maksimalverdiens indekser langs hver rad eller kolonne i tensoren ved å bruke ' dempet ' argument. Til slutt, vis maksimumsverdiens indeks i inngangstensoren. Denne bloggen har eksemplifisert metoden for å bruke 'torch.argmax()'-metoden i PyTorch.

Hvordan bruke 'torch.argmax()'-metoden i PyTorch?