Shortcuts

torch.nn.utils.vector_to_parameters

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

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

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

  • 参数 (可迭代[Tensor]) – 模型的参数,是一个Tensor的可迭代对象。