参数层

Pytorch 实现自定义参数层的例子

注意，一般官方接口都带有可导功能，如果你实现的层不具有可导功能，就需要自己实现梯度的反向传递。 官方Linear层： class Linear(Module): def __init__(se