LayerList

class paddle.nn. LayerList ( sublayers=None ) [源代码]

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

参数：

  • sublayers (iterable，可选) - 要保存的子层。

返回：无

代码示例

  1. import paddle
  2. import numpy as np
  4. class MyLayer(paddle.nn.Layer):
  5.     def __init__(self):
  6.         super(MyLayer, self).__init__()
  7.         self.linears = paddle.nn.LayerList(
  8.             [paddle.nn.Linear(10, 10) for i in range(10)])
  10.     def forward(self, x):
  11.         # LayerList can act as an iterable, or be indexed using ints
  12.         for i, l in enumerate(self.linears):
  13.             x = self.linears[i // 2](x) + l(x)
  14.         return x

append ( )

添加一个子层到整个list的最后。

参数：

  • sublayer (Layer) - 要添加的子层。

代码示例

  1. import paddle
  3. linears = paddle.nn.LayerList([paddle.nn.Linear(10, 10) for i in range(10)])
  4. another = paddle.nn.Linear(10, 10)
  5. linears.append(another)
  6. print(len(linears))  # 11

insert ( )

向list中插入一个子层，到给定的index前面。

参数：

  • index (int) - 要插入的位置。

  • sublayers (Layer) - 要插入的子层。

代码示例

  1. import paddle
  3. linears = paddle.nn.LayerList([paddle.nn.Linear(10, 10) for i in range(10)])
  4. another = paddle.nn.Linear(10, 10)
  5. linears.insert(3, another)
  6. print(linears[3] is another)  # True

extend ( )

添加多个子层到整个list的最后。

参数：

  • sublayers (iterable of Layer) - 要添加的所有子层。

代码示例

  1. import paddle
  3. linears = paddle.nn.LayerList([paddle.nn.Linear(10, 10) for i in range(10)])
  4. another_list = paddle.nn.LayerList([paddle.nn.Linear(10, 10) for i in range(5)])
  5. linears.extend(another_list)
  6. print(len(linears))  # 15
  7. print(another_list[0] is linears[10])  # True

使用本API的教程文档