neural network - autograd - Pytorch, ¿cuáles son los argumentos de gradiente?