torch.nn.utils.parameters_to_vector

torch.nn.utils.parameters_to_vector(parameters)

将参数的可迭代对象展平成一个单个向量。

参数:

参数（可迭代[Tensor]）- 一个包含模型参数的 Tensors 的可迭代对象。

返回值:

由单个向量表示的参数

返回类型:

张量