torch.nn.utils.vector_to_parameters

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

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

参数:

• vec（张量）- 表示模型参数的单个向量。

• 参数（可迭代[Tensor]）- 一个包含模型参数的 Tensors 的可迭代对象。