tf.experimental.numpy.argmax

TensorFlow variant of NumPy's argmax.

Unsupported arguments: out.

See the NumPy documentation for numpy.argmax.