tensorflow :: operaciones :: ArgMin

Devuelve el índice con el valor más pequeño en 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.argmin(input = a)
  c = tf.keras.backend.eval(b)
  # c = 0
  # here a[0] = 1 which is the smallest 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

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

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

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs OutputType(
  DataType x
)