排序、搜索和计数#
排序#
Return a sorted copy of a tensor. |
|
Sort a tensor, in-place. |
搜索#
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. |
|
Find the indices of tensor elements that are non-zero, grouped by element. |
|
Returns the indices of the minimum values along an axis. |
|
Return the indices of the elements that are non-zero. |
|
Return indices that are non-zero in the flattened version of a. |
|
Return elements, either from x or y, depending on condition. |
|
Find indices where elements should be inserted to maintain order. |
|
Return the elements of a tensor that satisfy some condition. |
计数#
Counts the number of non-zero values in the tensor |