tf.arg_min

tf.arg_min

tf.arg_min

arg_min(
    input,
    dimension,
    name=None
)

Defined in tensorflow/python/ops/gen_math_ops.py.

Returns the index with the smallest value across dimensions of a tensor.

Note that in case of ties the identity of the return value is not guaranteed.

Args:

• input: A Tensor. Must be one of the following types: float32, float64, int64, int32, uint8, uint16, int16, int8, complex64, complex128, qint8, quint8<
  登录查看完整内容