flujo tensor:: operaciones:: 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 empates, la identidad del valor devuelto no está garantizada.

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:

  • alcance: un objeto de alcance
  • dimensión: int32 o int64, debe estar en el rango [-rank(input), rank(input)) . Describe qué dimensión del tensor de entrada se 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:: operaciones:: ArgMax:: Atributos

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::Entrada 

 operator::tensorflow::Input() const

operador::tensorflow::Salida 

 operator::tensorflow::Output() const

Funciones estáticas públicas

Tipo de salida 

Attrs OutputType(
  DataType x
)