Shortcuts

torch.nn.utils.parameters_to_vector

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

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

Parameters

参数 (可迭代[Tensor]) – 模型的参数,是一个Tensor的可迭代对象。

Returns

由单个向量表示的参数

Return type

张量