Add argument to transform tensor to RAS (or any other orientation)