tensorflow:: ops:: ArgMax

#include <math_ops.h>

Devuelve el índice con el valor más grande en todas las dimensiones de un tensor.

Resumen

Tenga en cuenta que, en caso de empate, 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

Argumentos:

  • ámbito: Un objeto de ámbito
  • dimensión: int32 o int64, debe estar en el rango [-rank(input), rank(input)) . Describe qué dimensión del tensor de entrada se debe reducir. Para vectores, use dimensión = 0.

Devoluciones:

  • Output : El tensor de salida.

Constructores y destructores

ArgMax (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input dimension)
ArgMax (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input dimension, const ArgMax::Attrs & attrs)

Atributos públicos

operation
output

Funciones públicas

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

funciones estáticas públicas

OutputType (DataType x)

Estructuras

tensorflow:: ops:: ArgMax:: Attrs

Configuradores de atributos opcionales para ArgMax .

Atributos públicos

operación 

Operation operation

producción 

::tensorflow::Output output

Funciones públicas

ArgMax 

 ArgMax(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input dimension
)

ArgMax 

 ArgMax(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input dimension,
  const ArgMax::Attrs & attrs
)

nodo 

::tensorflow::Node * node() const

operador::tensorflow::Input 

 operator::tensorflow::Input() const

operador::tensorflow::Salida 

 operator::tensorflow::Output() const

funciones estáticas públicas

Tipo de salida 

Attrs OutputType(
  DataType x
)