Hi @ptrblck,
Yes, you are right, I was inputting a NumPy array. What I finally did was building Torchvision from the source, as the tensor use is only allowed in the master branch (not available in the releases).