UninitializedParameter¶
- class torch.nn.parameter.UninitializedParameter(requires_grad=True, device=None, dtype=None)[source][source]¶
一個未初始化的參數。
未初始化的參數是
torch.nn.Parameter
的一種特殊情況,其中資料的形狀仍然未知。與
torch.nn.Parameter
不同,未初始化的參數不包含任何資料,並且嘗試訪問某些屬性(如其形狀)將會引發執行階段錯誤。 可以對未初始化的參數執行的唯一操作是更改其資料類型、將其移動到不同的設備以及將其轉換為常規torch.nn.Parameter
。當參數被實體化時要使用的預設裝置或 dtype 可以在建構時設定,例如使用
device='cuda'
。