UninitializedBuffer

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

一個未初始化的緩衝區。

未初始化的緩衝區是 torch.Tensor 的一個特例，其中資料形狀未知。

與 torch.Tensor 不同，未初始化的引數不持有資料，嘗試訪問其某些屬性（例如形狀）將丟擲執行時錯誤。只能對未初始化的引數執行的操作包括：改變其資料型別、將其移動到不同的裝置以及將其轉換為常規的 torch.Tensor。

在例項化緩衝區時使用的預設裝置或資料型別可以在構造時透過例如 device='cuda' 進行設定。