Function dfdx::tensor_ops::sum_axis

source · [−]
pub fn sum_axis<T: Reduce1<I>, const I: isize>(t: T) -> T::Reduced
Expand description

Sum the values along dimension I of T.

Pytorch equivalent: t.sum(I)

Examples:

let t: Tensor2D<2, 3> = TensorCreator::zeros();
let a: Tensor1D<3> = t.clone().sum_axis::<0>();
let b: Tensor1D<2> = t.sum_axis::<-1>();