Tensorflow.JS - TF.argmax
Dans un tenseur, l'index commence par 0.
Syntaxe
TF.Argmax (Tensor_input, axe)Paramètres
1. Tensor_input est un tenseur qui a des éléments numériques.
Il peut être 1 ou 2 dimensions.
2. Si le tenseur est bidimensionnel, il est possible de spécifier l'axe pour obtenir un index de valeurs maximales dans les lignes ou les colonnes.
Si axe = 0, l'index des valeurs maximales est renvoyé sur le plan de la colonne, et si axe = 1, l'index des valeurs maximales est renvoyé sur la ligne.
Si l'axe n'est pas spécifié, il renverra l'index des valeurs maximales en termes de colonne.
Retour
Renvoie un tenseur avec les indices de valeur maximale.
Exemple 1
Créons un tenseur unidimensionnel dans JS qui a des valeurs entières et renvoie un index de valeur maximale.
Tensorflow.JS - TF.Argmax ()
Sortir
90 est le maximum parmi tous les éléments et est présent en 4e position. L'indice est 3.
Donc 3 est retourné.
Exemple 2
Créons un tenseur qui a 2 dimensions avec 4 lignes et 2 colonnes qui ont des valeurs entières et renvoie des indices de valeur maximale sur les colonnes.
Tensorflow.JS - TF.Argmax ()
Sortir
Fonctionnement
Tenseur du tenseur [[10, 13], [15, 6], [67, 5], [10, 2]]
La valeur maximale parmi (10,15,67,10) est de 67 et (13,6,5,2) est de 13.
Les positions d'index 67 et 13 sont 2 et 0.
Exemple 3
Créons un tenseur avec 2 dimensions en js avec 4 lignes et 2 colonnes avec des valeurs entières et renvoyer des indices de valeur maximale à travers les lignes.
Tensorflow.JS - TF.Argmax ()
Sortir
Fonctionnement
[[10, 13], [15, 6], [67, 5], [10, 2]]
Les valeurs maximales parmi [10,13] sont de 13, [15, 6] est de 15, [67, 5] est 67 et [10, 2] est 10.
Les positions d'index de 13 sont 1, 15 est 0, 67 est 0 et 10 est 0.
Exemple 4
Créons un tenseur avec 2 dimensions en js avec 4 lignes et 2 colonnes avec des valeurs entières et renvoyer les indices des valeurs maximales.
Tensorflow.JS - TF.Argmax ()
Sortir
Fonctionnement
Tensor [[1,2], [3,4], [5,6], [7,8]]
La valeur maximale dans la colonne [1,3,5,7] est de 7, et son indice est 3.
La valeur maximale dans la colonne [2,4,6,8] est de 8, et son indice est 3.
Conclusion
Dans ce Tensorflow.Tutoriel JS, nous avons vu comment renvoyer l'indice des éléments maximaux présents dans un tenseur en utilisant le TF.Méthode Argmax (). Dans un tenseur 2D, si axe = 0, l'index des valeurs maximales est renvoyé à travers la colonne, et si axe = 1, l'index des valeurs maximales est renvoyé sur la ligne. Par défaut, il renverra l'index des valeurs maximales en termes de colonne.