tf.keras.backend.argmin

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.