vector_to_parameters

paddle.nn.utils. vector_to_parameters ( vec, parameters, name=None ) [源代码]

将 1 个 1-D Tensor 按顺序切分给输入的多个 parameter。

参数

  • vec (Tensor) - 一个 1-D Tensor，它将被切片并复制到输入参数(input parameters)中。

  • parameters (Iterable[Tensor]) - 可迭代的多个 parameter。parameter 为 Layer 中可训练的 Tensor。

  • name (str，可选) - 具体用法请参见 Name，一般无需设置，默认值为 None。

返回

无

代码示例

>>> import paddle
>>> weight_attr = paddle.ParamAttr(initializer=paddle.nn.initializer.Constant(3.))
>>> linear1 = paddle.nn.Linear(10, 15, weight_attr)

>>> vec = paddle.nn.utils.parameters_to_vector(linear1.parameters())

>>> linear2 = paddle.nn.Linear(10, 15)
>>> # copy weight of linear1 to linear2
>>> paddle.nn.utils.vector_to_parameters(vec, linear2.parameters())
>>> print((linear1.weight == linear2.weight).all())
Tensor(shape=[], dtype=bool, place=Place(cpu), stop_gradient=True,
True)