UninitializedParameter¶
- class torch.nn.parameter.UninitializedParameter(requires_grad=True, device=None, dtype=None)[source][source]¶
一個未初始化的引數。
未初始化的引數是
torch.nn.Parameter的一種特殊情況,其資料形狀仍然未知。與
torch.nn.Parameter不同,未初始化的引數不持有資料,嘗試訪問其某些屬性(例如形狀)將丟擲執行時錯誤。對未初始化引數唯一可以執行的操作是更改其資料型別、將其移動到不同的裝置以及將其轉換為常規的torch.nn.Parameter。可以在構造時使用例如
device='cuda'來設定引數具體化時使用的預設裝置或資料型別。