Tune in to the first Women in ML Symposium this Tuesday, October 19 at 9am PST Register now

tf.keras.backend.argmax

TensorFlow 1 version View source on GitHub

Returns the index of the maximum value along an axis.

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

A tensor.