排序、搜索和计数#

排序#

maxframe.tensor.sort

返回 tensor 的排序副本。

maxframe.tensor.argsort

返回能够将 tensor 排序的索引。

maxframe.tensor.partition

返回 tensor 的分区副本。

maxframe.tensor.argpartition

使用由 kind 关键字指定的算法,沿给定轴执行间接分区。

搜索#

maxframe.tensor.argmax

返回沿轴的最大值的索引。

maxframe.tensor.nanargmax

返回沿指定轴忽略 NaN 后的最大值的索引。

maxframe.tensor.argmin

返回沿轴的最小值的索引。

maxframe.tensor.nanargmin

返回沿指定轴忽略 NaN 后的最小值的索引。

maxframe.tensor.nonzero

返回非零元素的索引。

maxframe.tensor.flatnonzero

返回在展平版本中非零的索引。

计数#

maxframe.tensor.argwhere

查找 tensor 中非零元素的索引,并按元素分组。

maxframe.tensor.count_nonzero

计算 tensor a 中非零值的个数。