Section courante

A propos

Section administrative du site

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



Dernière mise à jour : Samedi, le 12 Décembre 2020