Sorting, Searching, and Counting#
Sorting#
Return a sorted copy of a tensor. |
|
Returns the indices that would sort a tensor. |
|
Return a partitioned copy of a tensor. |
|
Perform an indirect partition along the given axis using the algorithm specified by the kind keyword. |
Searching#
Returns the indices of the maximum values along an axis. |
|
Return the indices of the maximum values in the specified axis ignoring NaNs. |
|
Returns the indices of the minimum values along an axis. |
|
Return the indices of the minimum values in the specified axis ignoring NaNs. |
|
Return the indices of the elements that are non-zero. |
|
Return indices that are non-zero in the flattened version of a. |
Counting#
Find the indices of tensor elements that are non-zero, grouped by element. |
|
Counts the number of non-zero values in the tensor |