mean

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

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

参数:

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

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

返回:

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

返回类型:

  • Variable(变量)。

代码示例

  1. import paddle.fluid as fluid
  2. import numpy
  3. # Graph Organizing
  4. input = fluid.layers.data(
  5. name='data', shape=[2, 3], dtype='float32')
  6. output = fluid.layers.mean(input)
  7. # Create an executor using CPU as an example
  8. place = fluid.CPUPlace()
  9. exe = fluid.Executor(place)
  10. exe.run(fluid.default_startup_program())
  11. # Execute
  12. x_ndarray = numpy.ones([2, 3]).astype(numpy.float32)
  13. res, = exe.run(fluid.default_main_program(),
  14. feed={'data':x_ndarray},
  15. fetch_list=[output])
  16. print(res)
  17. '''
  18. Output Value:
  19. [1.]
  20. '''