Retorna o índice com o maior valor entre as dimensões de um tensor.
Observe que em caso de empate a identidade do valor de retorno não é garantida.
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
Fragmento | OP_NAME | O nome desta operação, conforme conhecido pelo motor principal TensorFlow |
Métodos Públicos
Output <V> | asOutput () Retorna o identificador simbólico do tensor. |
estática <V estende TNumber > argmax <V> | |
estática argmax < TInt64 > | |
Output <V> | saída () |
Métodos herdados
Constantes
nome_op final String public static
O nome desta operação, conforme conhecido pelo motor principal TensorFlow
Métodos Públicos
pública Output <V> asOutput ()
Retorna o identificador simbólico do tensor.
As entradas para as operações do TensorFlow são saídas de outra operação do TensorFlow. Este método é usado para obter um identificador simbólico que representa o cálculo da entrada.
public static argmax <V> create ( Scope escopo, Operando <? estende TType > entrada, Operando <? estende TNumber > dimensão, Class <V> outputType)
Método de fábrica para criar uma classe envolvendo uma nova operação ArgMax.
Parâmetros
alcance | escopo atual |
---|---|
dimensão | int32 ou int64, deve estar no intervalo `[-rank (input), rank (input))`. Descreve qual dimensão do tensor de entrada deve ser reduzida. Para vetores, use dimensão = 0. |
Devoluções
- uma nova instância de ArgMax
public static argmax < TInt64 > create ( Scope escopo, Operando <? estende TType > entrada, Operando <? estende TNumber > dimensão)
Método de fábrica para criar uma classe envolvendo uma nova operação ArgMax usando tipos de saída padrão.
Parâmetros
alcance | escopo atual |
---|---|
dimensão | int32 ou int64, deve estar no intervalo `[-rank (input), rank (input))`. Descreve qual dimensão do tensor de entrada deve ser reduzida. Para vetores, use dimensão = 0. |
Devoluções
- uma nova instância de ArgMax