Devuelve el índice con el valor más grande en todas las dimensiones de un tensor.
Tenga en cuenta que en caso de empates, la identidad del valor devuelto no está garantizada.
Uso:
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
Cadena | OP_NOMBRE | El nombre de esta operación, como lo conoce el motor central de TensorFlow. |
Métodos públicos
Salida <V> | como salida () Devuelve el identificador simbólico del tensor. |
estático <V extiende TNumber > ArgMax <V> | |
estático ArgMax < TInt64 > | |
Salida <V> | producción () |
Métodos heredados
Constantes
Cadena final estática pública OP_NAME
El nombre de esta operación, como lo conoce el motor central de TensorFlow.
Métodos públicos
Salida pública <V> como Salida ()
Devuelve el identificador simbólico del tensor.
Las entradas a las operaciones de TensorFlow son salidas de otra operación de TensorFlow. Este método se utiliza para obtener un identificador simbólico que representa el cálculo de la entrada.
creación estática pública de ArgMax <V> (alcance alcance , operando <? extiende TType > entrada, operando <? extiende TNumber > dimensión, clase <V> tipo de salida)
Método de fábrica para crear una clase que envuelve una nueva operación ArgMax.
Parámetros
alcance | alcance actual |
---|---|
dimensión | int32 o int64, debe estar en el rango `[-rank(input), rank(input))`. Describe qué dimensión del tensor de entrada se reducirá. Para vectores, use dimensión = 0. |
Devoluciones
- una nueva instancia de ArgMax
ArgMax estático público <TInt64> crear (alcance alcance , operando <? extiende TType > entrada, operando <? extiende TNumber > dimensión)
Método de fábrica para crear una clase que envuelve una nueva operación ArgMax utilizando tipos de salida predeterminados.
Parámetros
alcance | alcance actual |
---|---|
dimensión | int32 o int64, debe estar en el rango `[-rank(input), rank(input))`. Describe qué dimensión del tensor de entrada se reducirá. Para vectores, use dimensión = 0. |
Devoluciones
- una nueva instancia de ArgMax