Function transpose 

pub fn transpose(input: &Tensor, perm: &[usize]) -> Result<Tensor>