torch.nn.utils.vector_to_parameters

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

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

参数

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

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