tf.keras.backend.argmax

TensorFlow 1 version 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.

Compat aliases