排序、搜索和计数#
排序#
返回 tensor 的排序副本。 |
|
返回能够将 tensor 排序的索引。 |
|
返回 tensor 的分区副本。 |
|
使用由 kind 关键字指定的算法,沿给定轴执行间接分区。 |
搜索#
返回沿轴的最大值的索引。 |
|
返回沿指定轴忽略 NaN 后的最大值的索引。 |
|
返回沿轴的最小值的索引。 |
|
返回沿指定轴忽略 NaN 后的最小值的索引。 |
|
返回非零元素的索引。 |
|
返回在展平版本中非零的索引。 |
计数#
查找 tensor 中非零元素的索引,并按元素分组。 |
|
计算 tensor |