Function autograd::ops::add_n

pub fn add_n(xs: &[&Tensor]) -> Tensor

Adds all input tensors.

All the input tensors must have same shapes.

extern crate ndarray;
extern crate autograd as ag;

let mut ctx = ag::Context::new();
let ref a = ag::ones(&[2, 2]);
let ref b = ag::ones(&[2, 2]);
let ref c = ag::ones(&[2, 2]);
let ref d = ag::add_n(&[a, b, c]);

assert_eq!(d.eval(&mut ctx).shape(), &[2, 2]);
assert_eq!(d.eval(&mut ctx), ndarray::arr2(&[[3., 3.], [3., 3.]]).into_dyn());