评价此页

torch.nn.utils.convert_parameters.vector_to_parameters#

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

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

参数
  • vec (Tensor) – representing the parameters of a model as a single vector.

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