Hi,
I think the answer here depends on the use case.
If you want to be use not to loose any precision, then yes you want to convert it to an int32 before converting to pytorch.
If you don’t want to use double the space, you can use int16 and it should work just fine as long as your numbers are not too big.
Do you encounter any issue with the choice made by torchvision?