评价此页

torch.nn.utils.parameters_to_vector#

torch.nn.utils.parameters_to_vector(parameters)[source]#

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

参数

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

返回

由单个向量表示的参数

返回类型

张量