Rate this Page
★ ★ ★ ★ ★

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.

On this page
Show Source
PyTorch Community
Language Bindings
PyTorch Libraries