Function autograd::ops::flatten

pub fn flatten(x: &Tensor) -> Tensor

Flattens input tensor into 1-ranked (vector)

extern crate autograd as ag;

let ref x = ag::zeros(&[3, 2, 2]);
let ref z = ag::flatten(x);
assert_eq!(z.eval(&[]).unwrap().shape(), &[12]);