mean

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

  • 参数:
    • x (Variable) : Tensor 或 LoDTensor。均值运算的输入。
    • name (basestring | None) : 输出变量的名称。
  • 返回:
    • Variable: 包含输出均值的 Tensor / LoDTensor。
  • 返回类型:
    • Variable(变量)。

代码示例

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