torch.nn.utils.vector_to_parameters

torch.nn.utils.vector_to_parameters(vec, parameters)[源代码]

将向量的切片复制到参数列表中。

参数

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

• parameters (Iterable[Tensor]) – 包含模型参数的张量集合。