Function autograd::ops::squeeze

pub fn squeeze<T: ArrayLike>(x: &Tensor, axes: &T) -> Tensor

Squeezes specified dims.

Each axis can be negative.

extern crate autograd as ag;

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

assert_eq!(b.eval(&mut ctx).shape(), &[3]);