LayerList

  • class paddle.fluid.dygraph.LayerList(sublayers=None)[源代码]

LayerList用于保存子层列表,它包含的子层将被正确地注册和添加。列表中的子层可以像常规python列表一样被索引。

  • 参数:
    • sublayers (iterable,可选) - 要保存的子层。

返回:无

代码示例

  1. import paddle.fluid as fluid
  2. import numpy as np
  3.  
  4. class MyLayer(fluid.Layer):
  5. def __init__(self):
  6. super(MyLayer, self).__init__()
  7. self.linears = fluid.dygraph.LayerList(
  8. [fluid.dygraph.Linear(10, 10) for i in range(10)])
  9. def forward(self, x):
  10. # LayerList可以像iterable一样迭代,也可以使用int索引
  11. for i, l in enumerate(self.linears):
  12. x = self.linears[i // 2](x) + l(x)
  13. return x