mean

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

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

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

  • name (basestring | None) : 输出变量的名称。

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

返回类型:
  • Variable(变量)。

代码示例

import paddle.fluid as fluid
import numpy

# Graph Organizing
input = fluid.layers.data(
    name='data', shape=[2, 3], dtype='float32')
output = fluid.layers.mean(input)

# Create an executor using CPU as an example
place = fluid.CPUPlace()
exe = fluid.Executor(place)
exe.run(fluid.default_startup_program())

# Execute
x_ndarray = numpy.ones([2, 3]).astype(numpy.float32)
res, = exe.run(fluid.default_main_program(),
               feed={'data':x_ndarray},
               fetch_list=[output])
print(res)
'''
Output Value:
[1.]
'''