Function autograd::ops::rank

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

Returns the (symbolic) rank of input tensor

extern crate ndarray;
extern crate autograd as ag;

let ref x = ag::zeros(&[2, 3, 4]);
let ref r = ag::rank(x);

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