UninitializedBuffer

class torch.nn.parameter.UninitializedBuffer(requires_grad=False, device=None, dtype=None, persistent=True)

一個尚未初始化的緩衝區。

未初始化緩衝區是 torch.Tensor 的一種特殊情況，其中資料的形狀仍然未知。

與 torch.Tensor 不同，未初始化的參數不包含任何資料，嘗試存取某些屬性（例如其形狀）將會拋出執行階段錯誤。唯一可以對未初始化參數執行的操作是更改其資料類型、將其移動到不同的裝置，以及將其轉換為常規的 torch.Tensor。

可以使用例如 device='cuda' 在建構期間設定緩衝區實體化時要使用的預設裝置或 dtype。