torch.nn.utils.vector_to_parameters

torch.nn.utils.vector_to_parameters(vec, parameters)

將一個向量的切片複製到引數的可迭代物件中。

引數

• vec (Tensor) – 代表模型引數的單個向量。

• parameters (Iterable[Tensor]) – 模型引數的 Tensor 的可迭代物件。