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:

Returns:

A tensor.