Function dfdx::tensor_ops::mean

source · [−]
pub fn mean<T: Reduce<Axes>, Axes>(t: T) -> T::Reducedwhere
    T::Array: HasAxes<Axes>,
Expand description

Average the values along Axes of T.

Pytorch equivalent: t.mean(Axes)

Example:

let t: Tensor2D<2, 3> = tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]]);
let r: Tensor0D = t.mean();
assert_eq!(r.data(), &3.5);

Reducing 1 axis:

let r: Tensor1D<2> = t.mean();
assert_eq!(r.data(), &[2.0, 5.0]);