tf.keras.backend.argmin

Returns the index of the minimum value along an axis.

x Tensor or variable.
axis axis along which to perform the reduction.

A tensor.