torch.nn.utils.
vector_to_parameters
Convert one vector to the parameters
vec (Tensor) – a single vector represents the parameters of a model.
parameters (Iterable[Tensor]) – an iterator of Tensors that are the parameters of a model.
此页内容是否对您有帮助
感谢反馈!