Retorna o índice com o maior valor nas dimensões de um tensor.
Observe que em caso de empate a identidade do valor retornado 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
Corda | OP_NAME | O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow |
Métodos Públicos
Saída <V> | asOutput () Retorna o identificador simbólico do tensor. |
estático <V estende TNumber > ArgMax <V> | |
ArgMax estático < TInt64 > | |
Saída <V> | saída () |
Métodos herdados
Constantes
String final estática pública OP_NAME
O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow
Métodos Públicos
Saída pública <V> asOutput ()
Retorna o identificador simbólico do tensor.
As entradas para 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 ( Escopo do escopo, Operando <? estende TType > entrada, Operando <? estende TNumber > dimensão, Class<V> outputType)
Método de fábrica para criar uma classe que envolve uma nova operação ArgMax.
Parâmetros
escopo | escopo atual |
---|---|
dimensão | int32 ou int64, deve estar no intervalo `[-rank(input), rank(input))`. Descreve qual dimensão do Tensor de entrada reduzir. Para vetores, use dimensão = 0. |
Devoluções
- uma nova instância de ArgMax
public static ArgMax < TInt64 > create ( Escopo do escopo, Operando <? estende TType > entrada, Operando <? estende TNumber > dimensão)
Método de fábrica para criar uma classe que envolve uma nova operação ArgMax usando tipos de saída padrão.
Parâmetros
escopo | escopo atual |
---|---|
dimensão | int32 ou int64, deve estar no intervalo `[-rank(input), rank(input))`. Descreve qual dimensão do Tensor de entrada reduzir. Para vetores, use dimensão = 0. |
Devoluções
- uma nova instância de ArgMax