tensorflow :: operaciones :: ArgMax

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.

Atributos públicos

Operation operation

::tensorflow::Output output

 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
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs OutputType(
  DataType x
)