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.]
'''