API

 torch / nn / torch.nn


torch.nn.utils.vector_to_parameters

torch.nn.utils.vector_to_parameters(vec: torch.Tensor, parameters: Iterable[torch.Tensor]) → None[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.


此页内容是否对您有帮助