Come utilizzare il metodo 'torch.argmax()' in PyTorch?

In PyTorch, il ' torcia.argmax() Il metodo ' è una funzione incorporata che restituisce indici di valori massimi di un particolare tensore su una data dimensione. Gli utenti utilizzano questa funzione quando lavorano con i tensori e desiderano trovare l'indice del valore massimo lungo la dimensione data di un tensore. Inoltre, questo metodo può essere utile anche per la classificazione in cui gli utenti vogliono sapere quale classe ha la probabilità più alta.

Questo blog esemplificherà il metodo per utilizzare il metodo 'torch.argmax()' in PyTorch.

Come utilizzare il metodo 'torch.argmax()' in PyTorch?

Il metodo 'torch.argmax()' prende come input qualsiasi tensore 1D o 2D e restituisce un tensore che contiene gli indici/indici dei valori massimi lungo la dimensione data.

La sintassi del metodo “torch.argmax()” è riportata di seguito:

torcia. argmax ( < input_tensore > )

Per utilizzare questo metodo in PyTorch, esamina i seguenti esempi per una migliore comprensione:

Esempio 1: utilizzare il metodo 'torch.argmax()' con tensore 1D

Nel primo esempio, creeremo un tensore 1D e con esso utilizzeremo il metodo “torch.argmax()”. Seguiamo la procedura passo passo riportata di seguito:

Passaggio 1: importa la libreria PyTorch

Innanzitutto, importa il ' torcia ' per utilizzare il metodo 'torch.argmax()':

importare torcia

Passaggio 2: creazione del tensore 1D

Quindi, crea un tensore 1D e stampa i suoi elementi. Qui, stiamo creando quanto segue “ Decine1 ' da un elenco utilizzando il tensore ' torcia.tensore() ' funzione:

Decine1 = torcia. tensore ( [ 5 , 0 , - 8 , 1 , 9 , 7 ] )

stampa ( Decine1 )

Questo ha creato un tensore 1D come mostrato di seguito:

Passaggio 3: trova gli indici di valore massimo

Ora, utilizza il ' torcia.argmax() ” per trovare l'indice/gli indici del valore massimo nel campo “ Decine1 ' tensore:

T1_ind = torcia. argmax ( Decine1 )

Passaggio 4: stampa dell'indice del valore massimo

Infine, visualizza l'indice del valore massimo nel tensore di input:

stampa ( 'Indici:' , T1_ind )

L'output seguente mostra l'indice del valore massimo nel ' Decine1 ' tensore cioè 4. Significa che il valore più alto del tensore è al 4° indice che è ' 9 ':

Esempio 2: utilizzare il metodo 'torch.argmax()' con tensore 2D

Nel secondo esempio, creeremo un tensore 2D e con esso utilizzeremo il metodo “torch.argmax()”. Seguiamo i passaggi forniti:

Passaggio 1: importa la libreria PyTorch

Innanzitutto, importa il ' torcia ' per utilizzare il metodo 'torch.argmax()':

importare torcia

Passaggio 2: crea tensore 2D

Quindi, utilizzare il ' torcia.tensore() ” per creare un tensore 2D e stamparne gli elementi. Qui, stiamo creando quanto segue “ Decine2 “Tensore 2D:

Decine2 = torcia. tensore ( [ [ 4 , 1 , - 7 ] , [ quindici , 6 , 0 ] , [ - 7 , 9 , 2 ] ] )

stampa ( Decine2 )

Questo ha creato un tensore 2D come mostrato di seguito:

Passaggio 3: trova gli indici di valore massimo

Ora, trova l’indice del valore massimo nel “ Decine2 ' tensore utilizzando il ' torcia.argmax() ' funzione:

T2_ind = torcia. argmax ( Decine2 )

Passaggio 4: stampa dell'indice del valore massimo

Infine, visualizza l’indice del valore massimo nel tensore di input:

stampa ( 'Indici:' , T2_ind )

Secondo l'output seguente, l'indice del valore massimo nel ' Decine2 ' il tensore è '3'. Significa che il valore più alto del tensore è al 3° indice che è “ quindici ':

Passaggio 5: trova gli indici di valore massimo lungo le colonne

Inoltre, lungo ciascuna colonna di un tensore, gli utenti possono anche trovare gli indici/indici dei valori massimi. Ad esempio, possiamo usare il ' debole=0 ' argomento con la funzione 'torch.argmax()'. Trova gli indici dei valori massimi lungo le colonne nella sezione “ Decine2 ' tensore e quindi stampa quegli indici:

indice_col = torcia. argmax ( Decine2 , debole = 0 )

stampa ( 'Indici in colonne:' , indice_col )

L'output seguente mostra gli indici dei valori massimi lungo ciascuna colonna del tensore:

Passaggio 6: trova gli indici di valore massimo lungo le righe

Allo stesso modo, gli utenti possono anche trovare gli indici/indici dei valori massimi lungo ciascuna riga di un tensore. Ad esempio, usa il ' debole=1 ' con la funzione 'torch.argmax()' per trovare gli indici dei valori massimi lungo le righe nel tensore 'Tens2' e quindi stampare tali indici:

indice_riga = torcia. argmax ( Decine2 , debole = 1 )

stampa ( 'Indici in righe:' , indice_riga )

Di seguito sono riportati gli indici del valore massimo lungo ciascuna riga di un tensore “Tens2”:

Abbiamo spiegato in modo efficiente il metodo per utilizzare il metodo “torch.argmax()” in PyTorch.

Nota : puoi accedere al nostro Google Colab Notebook da qui collegamento .

Conclusione

Per utilizzare il metodo 'torch.argmax()' in PyTorch, importa innanzitutto il file ' torcia ' biblioteca. Quindi, crea il tensore 1D o 2D desiderato e visualizza i suoi elementi. Successivamente, utilizzare il ' torcia.argmax() ” per trovare/calcolare gli indici/indici dei valori massimi nel tensore. Inoltre, gli utenti possono anche trovare gli indici del valore massimo lungo ciascuna riga o colonna nel tensore utilizzando il comando ' debole ' discussione. Infine, visualizza l'indice del valore massimo nel tensore di input. Questo blog ha esemplificato il metodo per utilizzare il metodo 'torch.argmax()' in PyTorch.

Come utilizzare il metodo 'torch.argmax()' in PyTorch?