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