UninitializedBuffer

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

一个未初始化的缓冲区。

未初始化的缓冲区是 torch.Tensor 的一个特例，其中数据形状未知。

与 torch.Tensor 不同，未初始化的参数不持有数据，尝试访问其某些属性（例如形状）将抛出运行时错误。只能对未初始化的参数执行的操作包括：改变其数据类型、将其移动到不同的设备以及将其转换为常规的 torch.Tensor。

在实例化缓冲区时使用的默认设备或数据类型可以在构造时通过例如 device='cuda' 进行设置。