Function autograd::ops::concat

pub fn concat(tensors: &[&Tensor], axis: isize) -> Tensor

Concatenates input tensors along specified axis.

axis can be negative.

extern crate autograd as ag;

let mut ctx = ag::Context::new();
let ref a = ag::zeros(&[3, 2]);
let ref b = ag::zeros(&[3, 2]);
let ref c = ag::zeros(&[3, 2]);
let ref d = ag::concat(&[a, b, c], 0);

assert_eq!(d.eval(&mut ctx).shape(), &[9, 2]);