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
booleano | es igual (Objeto arg0) |
Clase final<?> | obtenerclase () |
entero | código hash () |
vacío final | notificar () |
vacío final | notificar a todos () |
Cadena | a cadena () |
vacío final | esperar (arg0 largo, int arg1) |
vacío final | espera (arg0 largo) |
vacío final | esperar () |
entorno de ejecución abstracto | entorno () Devuelve el entorno de ejecución en el que se creó esta operación. |
operación abstracta |
Salida abstracta <V extiende TNumber > | como salida () Devuelve el identificador simbólico del tensor. |
resumen V extiende TNumber | como Tensor () Devuelve el tensor en este operando. |
forma abstracta | |
Clase abstracta <V extiende TNumber > | tipo () Devuelve el tipo de tensor de este operando. |
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