I have a question about how to define a custom module in pytorch. In my case, I need to define my own forward and backward function. Some example codes show that the autograd.Function subclass should have forward and backward parts, and the autograd.Module subclass should have init and forward parts. but in this way, I found the custom backward function has not been called. Can you give me some advice about this question ?
Thank you !