torch.nn.utils.parameters_to_vector

torch.nn.utils.parameters_to_vector(parameters)

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

参数:

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

返回:

将参数表示为单个向量

返回类型:

张量