Pytorch tomó la decisión correcta al estandarizarse en índices de 64 bits con signo. Probablemente seguiría haciendo decisiones caso por caso para usar int32 si fuera una opción. Algunos viejos hábitos persisten.