tf.arg_min( input, dimension, output_type=tf.int64, name=None )
Defined in tensorflow/python/ops/gen_math_ops.py
.
Returns the index with the smallest value across dimensions of a tensor. (deprecated)
THIS FUNCTION IS DEPRECATED. It will be removed in a future version. Instructions for updating: Use argmin
instead
Note that in case of ties the identity of the return value is not guaranteed.
input
: A Tensor
. Must be one of the following types: float32
, float64
, int32
, uint8
, int16
, int8
, complex64
, int64
, qint8
, quint8
, qint32
, bfloat16
, uint16
, complex128
, half
, uint32
, uint64
.dimension
: A Tensor
. Must be one of the following types: int32
, int64
. int32 or int64, must be in the range [-rank(input), rank(input))
. Describes which dimension of the input Tensor to reduce across. For vectors, use dimension = 0.output_type
: An optional tf.DType
from: tf.int32, tf.int64
. Defaults to tf.int64
.name
: A name for the operation (optional).A Tensor
of type output_type
.
© 2018 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/arg_min