Function autograd::ops::concat [−] [src]

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

Concatenates input tensors along specified axis.

axis can be negative.

extern crate autograd as ag;

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(&[]).unwrap().shape(), &[9, 2]);