UninitializedBuffer

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

一个未初始化的缓冲区。

UninitializedBuffer 是 torch.Tensor 的一种特殊情况，其中数据的形状仍然未知。

与 torch.Tensor 不同，未初始化的参数不保存任何数据，尝试访问某些属性（如形状）将抛出运行时错误。唯一可以对未初始化的参数执行的操作是更改其数据类型、将其移动到不同的设备并将其转换为常规的 torch.Tensor。

当缓冲区物化时要使用的默认设备或 dtype 可以在构造期间设置，例如使用 device='cuda'。