自定义层

中文版 pytorch
#1

https://zh-v2.d2l.ai/chapter_deep-learning-computation/custom-layer.html

#2

你好。我发现 named_parameters() 方法会自动捕捉到 MyLinear 类中定义的 weight 和 bias 参数， 请问 PyTorch 是如何发现这些参数的呢？

#3

因为你使用了 pytorch的接口函数，比如 nn.Parameter nn.Linera nn.conv2d …

#4

猜测是Module的相关方法，会分析类中类型为Parameter类型的参数，把这些成员标记为该module的参数，key值就是成员名。

1 Like
#5

记得应该是在nn.Module里面的__setattr__里面有，会对这些东西自动分类并放到模型的OrderedDict里

#6

第一问，不知道对不对，请大家指正。
class sumyk(nn.Module):
def init(self,xi,xj):
super().init()
self.weight = nn.Parameter(torch.randn(xi.shape[1],xj.shape[0]))
def forward(self,xi,xj):
return torch.matmul(torch.matmul(xi,self.weight.data),xj)

第二问求助