"estou a dizer-te, homem, devias simplesmente deixar o pytorch. faz tudo em cuda que escreveste tu mesmo"