cupy.amax#

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

返回数组的最大值或沿指定轴的最大值。

注意

当至少一个元素为 NaN 时，对应的最大值将为 NaN。

参数:

  • a (cupy.ndarray) – 需要计算最大值的数组。

  • axis (int) – 沿哪个轴计算最大值。默认为展平后的数组。

  • out (cupy.ndarray) – 输出数组。

  • keepdims (bool) – 如果为 True，则保留被缩减的轴，其大小为一。

返回:

a 的最大值；如果指定了轴，则返回沿该轴的最大值。

返回类型:

cupy.ndarray

注意

使用 cuTENSOR 加速器时，对于包含一个或多个 NaN 元素的缩减轴，输出值可能会被折叠（collapsed）。

另请参阅

numpy.amax()