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
Métodos herdados
booleano | é igual (objeto arg0) |
aula final<?> | getClass () |
interno | código hash () |
vazio final | notificar () |
vazio final | notificar todos () |
Corda | toString () |
vazio final | espere (long arg0, int arg1) |
vazio final | espere (arg0 longo) |
vazio final | espere () |
ambiente de execução abstrato | env () Retorne o ambiente de execução em que esta operação foi criada. |
operação abstrata |
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