torch.nn.utils.vector_to_parameters

torch.nn.utils.vector_to_parameters(vec, parameters) [source]

Copy slices of a vector into an iterable of parameters.

Parameters

• vec (Tensor) – a single vector representing the parameters of a model.
• parameters (Iterable[Tensor]) – an iterable of Tensors that are the parameters of a model.