flujo tensor:: 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, 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.

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
)