tf.experimental.numpy.argmax

TensorFlow variant of NumPy's argmax.

Unsupported arguments: out, keepdims.

See the NumPy documentation for numpy.argmax.