cupy.nanmax#
- cupy.nanmax(a, axis=None, out=None, keepdims=False)[source]#
返回数组沿指定轴忽略 NaN 后的最大值。
当某个切片的所有元素都是 NaN 时,将引发
RuntimeWarning
并返回 NaN。- 参数:
a (cupy.ndarray) – 要计算最大值的数组。
axis (int) – 计算最大值所沿的轴。默认使用扁平化数组。
out (cupy.ndarray) – 输出数组。
keepdims (bool) – 如果为
True
,则保留轴,使其大小为一。
- 返回:
数组
a
的最大值,如果指定了轴,则返回沿该轴的最大值。- 返回类型:
警告
此函数可能会同步设备。
另请参见