argmax |
Maximum d'argument |
TensorFlow version 2 |
Python |
Syntaxe
tf.math.argmax(
input, axis=None, output_type=tf.dtypes.int64, name=None
)
|
Paramètres
Nom |
Description |
input |
Ce paramètre permet d'indiquer un Tensor. |
axis |
Ce paramètre permet d'indiquer un entier avec l'axe à réduire. La valeur par défaut est 0. |
output_type |
Ce paramètre permet d'indiquer un type de sortie optionnel (tf.int32 ou tf.int64). La valeur par défaut est tf.int64. |
name |
Ce paramètre optionnel permet d'indiquer un nom pour cette opération. |
Description
Cette fonction permet de retourner l'index avec la plus grande valeur sur les axes d'un tenseur.
Remarque
- Lorsqu'un cas d'identité se produit, la fonction renvoie le plus petit index.
Dernière mise à jour : Samedi, le 12 Décembre 2020