ndarray.argmax

ndarray.argmax(self, axis=None, out=None)

Return indices of the maximum values along the given axis. Refer to mxnet.numpy.argmax for full documentation.