tf.argmin(input, axis=None, name=None, dimension=None)
See the guide: Math > Sequence Comparison and Indexing
Returns the index with the smallest value across axes of a tensor.
input
: A Tensor
. Must be one of the following types: float32
, float64
, int64
, int32
, uint8
, uint16
, int16
, int8
, complex64
, complex128
, qint8
, quint8
, qint32
, half
.axis
: A Tensor
. Must be one of the following types: int32
, int64
. int32, 0 <= axis < rank(input). Describes which axis of the input Tensor to reduce across. For vectors, use axis = 0.name
: A name for the operation (optional).A Tensor
of type int64
.
Defined in tensorflow/python/ops/math_ops.py
.
© 2017 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/argmin