Function dfdx::tensor_ops::sum

pub fn sum<T: Reduce<Axes>, Axes>(t: T) -> T::Reduced

Sum values along axes Axes of T.

Pytorch equivalent: t.sum(Axes)

Example:

let t: Tensor3D<2, 3, 4> = TensorCreator::zeros();
let _: Tensor0D = t.sum();

Reducing 1 axis:

let _: Tensor2D<3, 4> = t.sum();

Reducing multiple axes:

let _: Tensor1D<4> = t.sum();