description: TensorFlow variant of NumPy's argmax.

tf.experimental.numpy.argmax

View source on GitHub

TensorFlow variant of NumPy's argmax.

tf.experimental.numpy.argmax(
    a, axis=None
)

Unsupported arguments: out.

See the NumPy documentation for numpy.argmax.