torch.nn.utils.vector_to_parameters

torch.nn.utils.vector_to_parameters(vec, parameters) [исходный код]

Копирует срезы вектора в итерируемую последовательность параметров.

Параметры:

• vec (Tensor) – единый вектор, представляющий параметры модели.
• parameters (Iterable[Tensor]) – итерируемая последовательность тензоров, являющихся параметрами модели.