Clase public final argmax
Devuelve el índice con el valor más grande en todas las dimensiones de un tensor.
Tenga en cuenta que en caso de empates, no se garantiza la identidad del valor devuelto.
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
| Cuerda | OP_NAME | El nombre de esta operación, como lo conoce el motor central de TensorFlow |
Métodos públicos
| Output <V> | asOutput () Devuelve el identificador simbólico del tensor. |
| estática <V se extiende TNumber > argmax <V> | |
| estática argmax < TInt64 > | |
| Output <V> | salida () |
Métodos heredados
Constantes
OP_NAME pública final static String
El nombre de esta operación, como lo conoce el motor central de TensorFlow
Valor constante: "argmax"
Métodos públicos
público de salida <V> asOutput ()
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.
public static argmax <V> crear ( Alcance alcance, operando <? extiende Ttype > entrada, operando <? extiende TNumber > dimensión, Clase <V> OutputType)
Método de fábrica para crear una clase que envuelva una nueva operación ArgMax.
Parámetros
| alcance | alcance actual |
|---|---|
| dimensión | int32 o int64, debe estar en el rango `[-rank (entrada), rango (entrada))`. Describe en qué dimensión del tensor de entrada se debe reducir. Para los vectores, use dimension = 0. |
Devoluciones
- una nueva instancia de ArgMax
public static argmax < TInt64 > crear ( Alcance alcance, operando <? extiende Ttype > entrada, operando <? extiende TNumber > dimensión)
Método de fábrica para crear una clase que envuelva una nueva operación ArgMax usando tipos de salida predeterminados.
Parámetros
| alcance | alcance actual |
|---|---|
| dimensión | int32 o int64, debe estar en el rango `[-rank (entrada), rango (entrada))`. Describe en qué dimensión del tensor de entrada se debe reducir. Para los vectores, use dimension = 0. |
Devoluciones
- una nueva instancia de ArgMax