cupy.nanargmax#

cupy.nanargmax(a, axis=None, dtype=None, out=None, keepdims=False)[source]#

返回指定轴上忽略 NaN 的最大值的索引。对于全 NaN 的切片，返回 -1。尚不支持传递子类，subok=True 仍不支持

参数：

  • a (cupy.ndarray) – 要计算 nanargmax 的数组。

  • axis (int) – 沿哪个轴查找最大值。默认情况下会展平 a。

返回值：

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

返回类型：

cupy.ndarray

注意

出于性能考虑，对于全 NaN 的切片，cupy.nanargmax 返回“超出范围的值”，而 numpy.nanargmax 会引发 ValueError。

另请参阅

numpy.nanargmax()