Sure, and that’s what I’ve been doing (apologies for being unclear). It’s just a shame that the built in transforms are all PIL (but I guess that’s just torchvision, not core pytorch?).
Anyway, that was just my late night ranting… I love PyTorch!