cupy.nanargmin#

cupy.nanargmin(a, axis=None, dtype=None, out=None, keepdims=False)[源码]#

返回忽略 NaN 值后指定轴上最小值的索引。对于全为 NaN 的切片,返回 -1。目前不支持子类,subok=True 仍不受支持。

参数:
  • a (cupy.ndarray) – 要进行 nanargmin 操作的数组。

  • axis (int) – 沿哪个轴查找最小值。a 默认会被展平。

返回:

忽略 NaN 值后,a 沿某个轴的最小值的索引。

返回类型:

cupy.ndarray

注意

出于性能原因,cupy.nanargmin 对于全为 NaN 的切片返回 out of range values,而 numpy.nanargmin 会引发 ValueError