Function autograd::ops::transpose
[−]
[src]
pub fn transpose<T: ArrayLike>(x: &Tensor, perm: &T) -> Tensor
Permutes dimensions.
extern crate autograd as ag; let mut ctx = ag::Context::new(); let ref a = ag::zeros(&[1, 2, 3, 4, 5]); let ref b = ag::transpose(a, &[4, 2, 3, 0, 1]); assert_eq!(b.eval(&mut ctx).shape(), &[5, 3, 4, 1, 2]);