TensorFlow 1 version | View source on GitHub |
Returns the index of the minimum value along an axis.
tf.keras.backend.argmin(
x, axis=-1
)
Arguments | |
---|---|
x
|
Tensor or variable. |
axis
|
axis along which to perform the reduction. |
Returns | |
---|---|
A tensor. |