pymoose.mean

pymoose.mean(x, axis=None, placement=None)

Mean-reduce an input tensor.

Parameters

• x – A tensor.

• axis – An optional dimension along which to mean-reduce the tensor. If None, mean-reduces the entire tensor and outputs a scalar tensor.

• placement – An optional Placement to pin this operation to.

Returns

The averaged input, with one or all dimensions reduced out.