UninitializedBuffer#
- class torch.nn.parameter.UninitializedBuffer(requires_grad=False, device=None, dtype=None, persistent=True)[source]#
一個未初始化的 buffer。
Uninitialized Buffer 是
torch.Tensor的一個特殊情況,其資料形狀仍未知。與
torch.Tensor不同,未初始化的引數不包含資料,嘗試訪問其某些屬性(例如形狀)將引發執行時錯誤。可以在未初始化的引數上執行的唯一操作是更改其資料型別、將其移動到不同的裝置以及將其轉換為常規的torch.Tensor。當 buffer 被具體化時使用的預設裝置或 dtype 可以在構造期間設定,例如使用
device='cuda'。