O Pytorch tomou a decisão certa ao padronizar índices de 64 bits com sinal. Eu provavelmente ainda estaria a fazer decisões caso a caso para usar int32 se fosse uma opção. Alguns velhos hábitos persistem.