Sorting, Searching, and Counting

Searching

mars.tensor.argmax Returns the indices of the maximum values along an axis.
mars.tensor.nanargmax Return the indices of the maximum values in the specified axis ignoring NaNs.
mars.tensor.argmin Returns the indices of the minimum values along an axis.
mars.tensor.nanargmin Return the indices of the minimum values in the specified axis ignoring NaNs.
mars.tensor.argwhere Find the indices of tensor elements that are non-zero, grouped by element.
mars.tensor.argmin Returns the indices of the minimum values along an axis.
mars.tensor.nonzero
mars.tensor.flatnonzero
mars.tensor.where Return elements, either from x or y, depending on condition.
mars.tensor.extract

Counting

mars.tensor.count_nonzero Counts the number of non-zero values in the tensor a.