評價此頁

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'