mean

paddle.fluid.layers. mean ( x, name=None ) [源代码]

计算 x 所有元素的平均值。

参数

  • x (Variable) : Tensor 或 LoDTensor。均值运算的输入。

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

返回

  • Variable:包含输出均值的 Tensor / LoDTensor。

返回类型

  • Variable(变量)。

代码示例

import paddle
import paddle.fluid as fluid
paddle.enable_static()

input = fluid.layers.data(
    name='data', shape=[2, 3], dtype='float32')
mean = paddle.mean(input)