torch.nn.utils.vector_to_parameters

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

Convert one vector to the parameters

Parameters

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

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.nn.utils.vector_to_parameters.html