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 Return the indices of the elements that are non-zero.
mars.tensor.flatnonzero Return indices that are non-zero in the flattened version of a.
mars.tensor.where Return elements, either from x or y, depending on condition.
mars.tensor.extract Return the elements of a tensor that satisfy some condition.

Counting

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