pytorch自定义扩展

一 torch.nn.Module和torch.autograd.Function

pytorch的自定义拓展之（一）------torch.nn.Module和torch.autograd.Function_用torch.autograd.function还是nn.module-CSDN博客https://blog.csdn.net/qq_27825451/article/details/95189376