Pytorch made the right call standardizing on signed 64 bit indexes. I would probably still be rather pointlessly making case by case decisions to use int32 if it were an option. Some old habits linger.
PyTorch’s Signed 64-bit Index Standardization Decision
By
–