快捷方式

torch.nn.utils.parameters_to_vector

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

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

引數

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

返回

表示引數的單個向量

返回型別

Tensor


© 版權所有 PyTorch 貢獻者。

使用由 Read the Docs 提供的 Sphinx 主題構建。

文件

訪問 PyTorch 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源