tf.keras.ops.argmin

Returns the indices of the minium values along an axis.

View aliases

Main aliases

tf.keras.ops.numpy.argmin

tf.keras.ops.argmin(
    x, axis=None, keepdims=False
)

Args
x	Input tensor.
axis	By default, the index is into the flattened tensor, otherwise along the specified axis.
keepdims	If this is set to True, the axes which are reduced are left in the result as dimensions with size one. Defaults to False.

Returns
Tensor of indices. It has the same shape as x, with the dimension along axis removed.

Example:

x = keras.ops.arange(6).reshape(2, 3) + 10
x
array([[10, 11, 12],
       [13, 14, 15]], dtype=int32)
keras.ops.argmin(x)
array(0, dtype=int32)
keras.ops.argmin(x, axis=0)
array([0, 0, 0], dtype=int32)
keras.ops.argmin(x, axis=1)
array([0, 0], dtype=int32)