torch.nn.utils.parameters_to_vector

torch.nn.utils.parameters_to_vector(parameters)[源代码]

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

参数

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

返回值

用一个向量表示的参数

返回类型

Tensor