tensorflow :: 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, 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:

  • alcance: un objeto de alcance
  • dimensión: int32 o int64, debe estar en el rango [-rank(input), rank(input)) . Describe en qué dimensión del tensor de entrada se debe reducir. Para los vectores, use dimension = 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 publicas

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

Establecedores de atributos opcionales para ArgMax .

Atributos públicos

operación

Operation operation

producción

::tensorflow::Output output

Funciones publicas

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
)