評價此頁

torch.nn.utils.parameters_to_vector#

torch.nn.utils.parameters_to_vector(parameters)[原始碼]#

將引數可迭代物件展平成單個向量。

引數

parameters (Iterable[Tensor]) – 模型引數的 Tensor 的可迭代物件。

返回

將引數表示為單個向量

返回型別

張量