7.5.36.4. argmax

Array.argmax(axis=None, out=None, *, keepdims=False)

Return indices of the maximum values along the given axis.

Refer to numpy.argmax for full documentation.

7.5.36.4. See Also

numpy.argmax : equivalent function