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.

Usage: 

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

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.
Retours
  • 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.
Retours
  • une nouvelle instance d'ArgMax

sortie publique <V> sortie ()