自定义层

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

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

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