Sorting, Searching, and Counting#

Searching#

maxframe.tensor.argmax

Returns the indices of the maximum values along an axis.

maxframe.tensor.nanargmax

Return the indices of the maximum values in the specified axis ignoring NaNs.

maxframe.tensor.argmin

Returns the indices of the minimum values along an axis.

maxframe.tensor.nanargmin

Return the indices of the minimum values in the specified axis ignoring NaNs.

Counting#

maxframe.tensor.count_nonzero

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