Function autograd::ops::size
[−]
[src]
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(&[])]);