快捷鍵

torch.nn.utils.parameters_to_vector

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

將參數的可迭代對象展平為單一向量。

參數

parameters (Iterable[Tensor]) – 一個 Tensor 的可迭代對象,這些 Tensor 是模型的參數。

返回

由單一向量表示的參數

返回類型

Tensor

文件

取得 PyTorch 的完整開發者文件

檢視文件

教學

取得適合初學者和進階開發者的深入教學

檢視教學

資源

尋找開發資源並取得您的問題解答

檢視資源