vector_to_parameters

paddle.nn.utils. vector_to_parameters ( vec, parameters, name=None ) [source]

Transform a 1-D Tensor to the input parameters .

Parameters
  • vec (Tensor) – A 1-D Tensor, which will be sliced and copied to the input parameters .

  • parameters (Iterable[Tensor]) – Iterable Tensors that are trainable parameters of a Layer.

  • name (str, optional) – The default value is None. Normally there is no need for user to set this property. For more information, please refer to Name.

Examples

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())
# weight: Tensor(shape=[10, 15], dtype=float32, place=CUDAPlace(0), stop_gradient=False,
#                 [[3. , ..., 3. ],
#                  [..., ..., ...],
#                  [3. , ..., 3. ]])