Hi,
You cannot.
To be able to use the autograd, you need to use pytorch’s function (because they contain the information on how to compute the gradients). You cannot use numpy operations and backpropagate through them.