tf.keras.backend.argmax

View source on GitHub

Returns the index of the maximum value along an axis.

tf.keras.backend.argmax(
    x, axis=-1
)

Arguments:

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

Returns:

A tensor.