max

paddle. max ( x, axis=None, keepdim=False, name=None )

对指定维度上的 Tensor 元素求最大值运算，并输出相应的计算结果。

注解

对输入有多个最大值的情况下，max 将梯度完整传回到最大值对应的位置，amax 会将梯度平均传回到最大值对应的位置

参数

  • x (Tensor) - Tensor，支持数据类型为 float8_e4m3fn, float8_e5m2, bfloat16, float16, float32，float64，int32，int64。

  • axis (list|int|tuple，可选) - 求最大值运算的维度。如果为 None，则计算所有元素的最大值并返回包含单个元素的 Tensor 变量，否则必须在 \([-x.ndim, x.ndim]\) 范围内。如果 \(axis[i] <0\)，则维度将变为 \(x.ndim+axis[i]\)，默认值为 None。

  • keepdim (bool，可选) - 是否在输出 Tensor 中保留输入的维度。除非 keepdim 为 True，否则输出 Tensor 的维度将比输入 Tensor 小一维，默认值为 False。

  • name (str，可选) - 具体用法请参见 Name，一般无需设置，默认值为 None。

返回

Tensor，在指定 axis 上进行求最大值运算的 Tensor，数据类型和输入数据类型一致。

代码示例

COPY-FROM: paddle.max

使用本API的教程文档