Function autograd::ops::transpose

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]);