argmax

ivy.argmax(x, axis=0, f=None)[source]

Returns the index with the largest value across axes of an array.

Parameters
  • x (array) – Input array containing elements to argmax.

  • axis (int, optional) – Axis to perform the argmax, default is 0.

  • f (ml_framework, optional) – Machine learning framework. Inferred from inputs if None.

Returns

Tensor containing the indices of the maximum values across the specified axis.


Supported Frameworks:

empty jax_logo empty tf_logo empty pytorch_logo empty mxnet_logo empty numpy_logo empty