"te lo digo, amigo, deberías dejar PyTorch. hazlo todo en CUDA que has escrito tú mismo"