Ce blog illustrera la méthode pour utiliser la méthode « torch.argmax() » dans PyTorch.
Comment utiliser la méthode « torch.argmax() » dans PyTorch ?
La méthode « torch.argmax() » prend n'importe quel tenseur 1D ou 2D en entrée et renvoie un tenseur qui contient les indices/index des valeurs maximales le long de la dimension donnée.
La syntaxe de la méthode « torch.argmax() » est donnée ci-dessous :
torche. argmax ( < tenseur_entrée > )
Pour utiliser cette méthode dans PyTorch, parcourez les exemples suivants pour une meilleure compréhension :
Exemple 1 : utiliser la méthode « torch.argmax() » avec un tenseur 1D
Dans le premier exemple, nous allons créer un tenseur 1D et utiliser la méthode « torch.argmax() » avec lui. Suivons la procédure étape par étape ci-dessous :
Étape 1 : Importer la bibliothèque PyTorch
Tout d’abord, importez le « torche ' pour utiliser la méthode 'torch.argmax()' :
importer torcheÉtape 2 : Créer un tenseur 1D
Ensuite, créez un tenseur 1D et imprimez ses éléments. Ici, nous créons ce qui suit » Dizaines1 ' tenseur d'une liste en utilisant le ' torche.tensor() ' fonction:
Dizaines1 = torche. tenseur ( [ 5 , 0 , - 8 , 1 , 9 , 7 ] )
imprimer ( Dizaines1 )
Cela a créé un tenseur 1D comme indiqué ci-dessous :
Étape 3 : Trouver les indices de valeur maximale
Maintenant, utilisez le ' torche.argmax() ' Fonction pour trouver le ou les index de la valeur maximale dans le ' Dizaines1 ' Tenseur :
T1_ind = torche. argmax ( Dizaines1 )Étape 4 : Imprimer l'index de la valeur maximale
Enfin, affichez l'index de la valeur maximale dans le tenseur d'entrée :
imprimer ( 'Indices :' , T1_ind )La sortie ci-dessous montre l'index de la valeur maximale dans le ' Dizaines1 ' tenseur c'est-à-dire 4. Cela signifie que la valeur la plus élevée du tenseur est au 4ème indice qui est ' 9 » :
Exemple 2 : Utiliser la méthode « torch.argmax() » avec un tenseur 2D
Dans le deuxième exemple, nous allons créer un tenseur 2D et utiliser la méthode « torch.argmax() » avec lui. Suivons les étapes fournies :
Étape 1 : Importer la bibliothèque PyTorch
Tout d’abord, importez le « torche ' pour utiliser la méthode 'torch.argmax()' :
importer torcheÉtape 2 : Créer un tenseur 2D
Ensuite, utilisez le ' torche.tensor() » pour créer un tenseur 2D et imprimer ses éléments. Ici, nous créons ce qui suit » Dizaines2 « Tenseur 2D :
Dizaines2 = torche. tenseur ( [ [ 4 , 1 , - 7 ] , [ quinze , 6 , 0 ] , [ - 7 , 9 , 2 ] ] )imprimer ( Dizaines2 )
Cela a créé un tenseur 2D comme indiqué ci-dessous :
Étape 3 : Trouver les indices de valeur maximale
Maintenant, trouvez l'indice de la valeur maximale dans le champ ' Dizaines2 ' tenseur en utilisant le ' torche.argmax() ' fonction:
T2_ind = torche. argmax ( Dizaines2 )Étape 4 : Imprimer l'index de la valeur maximale
Enfin, affichez l'index de la valeur maximale dans le tenseur d'entrée :
imprimer ( 'Indices :' , T2_ind )Selon la sortie ci-dessous, l'indice de la valeur maximale dans le ' Dizaines2 ' Le tenseur est '3'. Cela signifie que la valeur la plus élevée du tenseur se situe au 3ème indice qui est « quinze » :
Étape 5 : Trouver les indices de valeur maximale le long des colonnes
De plus, les utilisateurs peuvent également trouver les indices/indices des valeurs maximales le long de chaque colonne d'un tenseur. Par exemple, nous pouvons utiliser le « faible=0 » avec la fonction « torch.argmax() ». Il trouve les indices des valeurs maximales le long des colonnes dans le « Dizaines2 ' tenseur puis imprime ces indices :
col_index = torche. argmax ( Dizaines2 , faible = 0 )imprimer ( 'Indices en colonnes :' , col_index )
La sortie ci-dessous montre les index des valeurs maximales le long de chaque colonne du tenseur :
Étape 6 : Trouver les indices de valeur maximale le long des lignes
De même, les utilisateurs peuvent également trouver les indices/indices des valeurs maximales le long de chaque ligne d'un tenseur. Par exemple, utilisez le « faible=1 ' avec la fonction 'torch.argmax()' pour trouver les indices des valeurs maximales le long des lignes du tenseur 'Tens2', puis imprimer ces indices :
index_ligne = torche. argmax ( Dizaines2 , faible = 1 )imprimer ( 'Indices en lignes :' , index_ligne )
Les index de la valeur maximale le long de chaque ligne d'un tenseur « Tens2 » sont visibles ci-dessous :
Nous avons expliqué efficacement la méthode pour utiliser la méthode « torch.argmax() » dans PyTorch.
Note : Vous pouvez accéder à notre bloc-notes Google Colab à cette adresse lien .
Conclusion
Pour utiliser la méthode « torch.argmax() » dans PyTorch, importez d’abord le « torche ' bibliothèque. Ensuite, créez le tenseur 1D ou 2D souhaité et visualisez ses éléments. Ensuite, utilisez le ' torche.argmax() » pour trouver/calculer les indices/indices des valeurs maximales dans le tenseur. De plus, les utilisateurs peuvent également trouver les index de la valeur maximale le long de chaque ligne ou colonne du tenseur en utilisant le « faible ' argument. Enfin, affichez l’index de la valeur maximale dans le tenseur d’entrée. Ce blog a illustré la méthode pour utiliser la méthode « torch.argmax() » dans PyTorch.