LayerList
- class
paddle.fluid.dygraph.
LayerList
(sublayers=None)[源代码]
LayerList用于保存子层列表,它包含的子层将被正确地注册和添加。列表中的子层可以像常规python列表一样被索引。
- 参数:
- sublayers (iterable,可选) - 要保存的子层。
返回:无
代码示例
- import paddle.fluid as fluid
- import numpy as np
- class MyLayer(fluid.Layer):
- def __init__(self):
- super(MyLayer, self).__init__()
- self.linears = fluid.dygraph.LayerList(
- [fluid.dygraph.Linear(10, 10) for i in range(10)])
- def forward(self, x):
- # LayerList可以像iterable一样迭代,也可以使用int索引
- for i, l in enumerate(self.linears):
- x = self.linears[i // 2](x) + l(x)
- return x