Unfortunately some of the torchvision transforms have some limitations to what dimensionality they can handle (namely transforms intended for PIL Images). However if you just wish to normalize the data, you can use transforms.Normalize() on any N dim data given that you convert the data to a Torch Tensor in your data loader.