Function autograd::ops::flatten

pub fn flatten<T: Float, A: AsRef<Tensor<T>>>(x: A) -> Tensor<T>

Flattens input tensor into 1-ranked (vector).

extern crate autograd as ag;

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