在__init__外部添加子层,还是不在Tensorflow 2中直接添加子层?

时间:2019-11-05 02:35:36

标签: tensorflow tensorflow2.0

Here,其中提到以下内容:

  

如果您将Layer实例指定为   另一层的属性,外层将开始跟踪   内层的重量。

     

我们建议使用__init__方法创建此类子层(因为   子层通常具有构建方法,它们将在   外层被构建)。

在我的用例中,我在__init__中创建子层,但不直接在self中注册它们。相反,我将它们保留在辅助对象中,然后将其存储在self中。我已经在PyTorch中实现了类似的代码,对于这种情况,如果不是直接在self中创建的,则必须通过add_module手动添加图层(即PyTorch模块)。 Tensorflow 2也是如此吗?如果是这样,对应的呼叫是什么?

0 个答案:

没有答案