Sorting, Searching, and Counting#

Sorting#

mars.tensor.sort

Return a sorted copy of a tensor.

mars.tensor.core.Tensor.sort

Sort a tensor, in-place.

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.searchsorted

Find indices where elements should be inserted to maintain order.

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.