maxframe.tensor.nanargmax#

maxframe.tensor.nanargmax(a, axis=None, out=None)[源代码]#

返回指定轴上忽略 NaN 的最大值的索引。对于全为 NaN 的切片会引发 ValueError。警告：如果切片只包含 NaN 和 -Inf，则结果不可信。

参数:

  • a (array_like) -- 输入数据。

  • axis (int, optional) -- 运算的轴。默认使用展平后的输入。

  • out (Tensor, optional) -- 用于存放结果的替代输出张量。默认为 None；如果提供，它必须与期望输出具有相同的形状，但类型会在必要时进行转换。详见 doc.ufuncs。

返回:

index_array -- 一个索引张量或单个索引值。

返回类型:

Tensor

参见

argmax, nanargmin

示例

>>> import maxframe.tensor as mt

>>> a = mt.array([[mt.nan, 4], [2, 3]])
>>> mt.argmax(a).execute()
0
>>> mt.nanargmax(a).execute()
1
>>> mt.nanargmax(a, axis=0).execute()
array([1, 0])
>>> mt.nanargmax(a, axis=1).execute()
array([1, 1])