Function autograd::ops::size

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

Returns the (symbolic) size of input tensor

extern crate ndarray;
extern crate autograd as ag;

let ref a = ag::zeros(&[4, 3]);
let ref b = ag::size(a);

let mut ctx = ag::Context::new();
assert_eq!(12., b.eval(&mut ctx)[ndarray::IxDyn(&[])]);