ParameterList

class paddle.nn. ParameterList ( parameters=None ) [源代码] 

参数列表容器。此容器的行为类似于 Python 列表，但它包含的参数将被正确地注册和添加。

参数

parameters (iterable，可选) - 可迭代的 Parameters。

返回

无

代码示例

          >>> import paddle

>>> class MyLayer(paddle.nn.Layer):
...     def __init__(self, num_stacked_param):
...         super().__init__()
...         # create ParameterList with iterable Parameters
...         self.params = paddle.nn.ParameterList(
...             [paddle.create_parameter(shape=[2, 2], dtype='float32') for _ in range(num_stacked_param)]
...         )
...
...     def forward(self, x):
...         for i, p in enumerate(self.params):
...             x = paddle.matmul(x, p)
...         return x
>>> x = paddle.uniform(shape=[5, 2], dtype='float32')
>>> num_stacked_param = 4
>>> model = MyLayer(num_stacked_param)
>>> print(len(model.params))
4
>>> res = model(x)
>>> print(res.shape)
paddle.Size([5, 2])
>>> replaced_param = paddle.create_parameter(shape=[2, 3], dtype='float32')
>>> model.params[num_stacked_param - 1] = replaced_param
>>> res = model(x)
>>> print(res.shape)
paddle.Size([5, 3])
>>> model.params.append(paddle.create_parameter(shape=[3, 4], dtype='float32'))  # append param
>>> print(len(model.params))
5
>>> res = model(x)
>>> print(res.shape)
paddle.Size([5, 4])

         