Function autograd::ops::squeeze
[−]
[src]
pub fn squeeze<T: ArrayLike>(x: &Tensor, axes: &T) -> Tensor
Squeezes specified dims.
Each axis can be negative.
extern crate autograd as ag; let ref a = ag::zeros(&[1, 3, 1]); let ref b = ag::squeeze(a, &[0, 2]); assert_eq!(b.eval(&[]).unwrap().shape(), &[3]);