Here,其中提到以下内容:
如果您将Layer实例指定为 另一层的属性,外层将开始跟踪 内层的重量。
我们建议使用
__init__
方法创建此类子层(因为 子层通常具有构建方法,它们将在 外层被构建)。
在我的用例中,我在__init__
中创建子层,但不直接在self
中注册它们。相反,我将它们保留在辅助对象中,然后将其存储在self
中。我已经在PyTorch中实现了类似的代码,对于这种情况,如果不是直接在self
中创建的,则必须通过add_module
手动添加图层(即PyTorch模块)。 Tensorflow 2也是如此吗?如果是这样,对应的呼叫是什么?