torch.nn.utils.vector_to_parameters

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

将一个向量的切片复制到参数的可迭代对象中。

参数:

• vec (Tensor) – 一个表示模型参数的单一向量。

• parameters (Iterable[Tensor]) – 模型参数的 Tensor 的可迭代对象。