ptrblck:

In


thanks, i removed @staticmethod in my code then i am facing below error
Legacy autograd function with non-static forward method is deprecated. Please use new-style autograd function with static forward method. (Example: https://pytorch.org/docs/stable/autograd.html#torch.autograd.Function  21).