torch.nn.utils.vector_to_parameters

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

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

参数

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

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