tochvision.transforms use PIL under the hood which doesn't work with 3D data. It also helps you to understand what's going on with your data if you're used to using numpy.