Custom backward for a NN with non linear activation function

[CS_PRANEETH]

how should the custom backward() be defined when i have a non linear activation function?

I think in the docw he gives an example where there the neural networks has no activation function.
https://pytorch.org/docs/master/notes/extending.html