ArgMax

classe finale publique ArgMax

Renvoie l'index avec la plus grande valeur sur toutes les dimensions d'un tenseur.

Notez qu'en cas d'égalité, l'identité de la valeur de retour n'est pas garantie.

Utilisation :

import tensorflow as tf
   a = [1, 10, 26.9, 2.8, 166.32, 62.3]
   b = tf.math.argmax(input = a)
   c = tf.keras.backend.eval(b)
   # c = 4
   # here a[4] = 166.32 which is the largest element of a across axis 0
   

Constantes

Chaîne

OP_NAME

Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Méthodes publiques

Sortie <V>	comme Sortie () Renvoie le handle symbolique du tenseur.
statique <V étend TNumber > ArgMax <V>	créer ( Scope scope, Operand <? extends TType > input, Operand <? extends TNumber > dimension, Class<V> outputType) Méthode d'usine pour créer une classe encapsulant une nouvelle opération ArgMax.
ArgMax statique < TInt64 >	créer (Portée de la portée , Opérande <? étend TType > entrée, Opérande <? étend TNumber > dimension) Méthode d'usine pour créer une classe encapsulant une nouvelle opération ArgMax à l'aide des types de sortie par défaut.
Sortie <V>	sortir ()

Méthodes héritées

De la classe org.tensorflow.op.RawOp

booléen final	est égal (Obj objet)
entier final	Code de hachage ()
Opération	op () Renvoie cette unité de calcul sous la forme d'une seule Operation .
Chaîne finale	àChaîne ()

De la classe java.lang.Object

booléen	est égal à (Objet arg0)
Classe finale<?>	obtenirClasse ()
int	Code de hachage ()
vide final	notifier ()
vide final	notifierTous ()
Chaîne	àChaîne ()
vide final	attendre (long arg0, int arg1)
vide final	attendez (long arg0)
vide final	attendez ()

Depuis l'interface org.tensorflow.op.Op

résuméExécutionEnvironnement	env () Renvoie l'environnement d'exécution dans lequel cette opération a été créée.
opération abstraite	op () Renvoie cette unité de calcul sous la forme d'une seule Operation .

Depuis l'interface org.tensorflow.Operand

sortie abstraite <V étend TNumber >	comme Sortie () Renvoie le handle symbolique du tenseur.
abstrait V étend TNumber	asTensor () Renvoie le tenseur à cet opérande.
Forme abstraite	forme () Renvoie la forme (éventuellement partiellement connue) du tenseur auquel fait référence la Output de cet opérande.
Classe abstraite <V étend TNumber >	taper () Renvoie le type de tenseur de cet opérande

Depuis l'interface org.tensorflow.ndarray.Shaped

abstrait entier	rang ()
Forme abstraite	forme ()
abstrait long	taille () Calcule et renvoie la taille totale de ce conteneur, en nombre de valeurs.

Constantes

chaîne finale statique publique OP_NAME

Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Valeur constante : "ArgMax"

Méthodes publiques

sortie publique <V> asOutput ()

Renvoie le handle symbolique du tenseur.

Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.

public static ArgMax <V> créer ( Scope scope, Operand <? extends TType > input, Operand <? extends TNumber > dimension, Class<V> outputType)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération ArgMax.

Paramètres

portée	portée actuelle
dimension	int32 ou int64, doit être compris dans la plage `[-rank(input), Rank(input))`. Décrit la dimension du Tensor d'entrée à réduire. Pour les vecteurs, utilisez dimension = 0.

Retour

• une nouvelle instance d'ArgMax

public static ArgMax < TInt64 > create ( Scope scope, Operand <? extends TType > input, Operand <? extends TNumber > dimension)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération ArgMax à l'aide des types de sortie par défaut.

Paramètres

portée	portée actuelle
dimension	int32 ou int64, doit être compris dans la plage `[-rank(input), Rank(input))`. Décrit la dimension du Tensor d'entrée à réduire. Pour les vecteurs, utilisez dimension = 0.

Retour

• une nouvelle instance d'ArgMax

sortie publique <V> sortie ()