dpnp.ndarray.argmax

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

Returns array of indices of the maximum values along the given axis.

Refer to dpnp.argmax for full documentation.