未初始化的缓冲区 ¶

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

未初始化的缓冲区。

未初始化的缓冲区是 torch.Tensor 的一种特殊情况，其中数据形状仍然未知。

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

当缓冲区实例化时，可以设置默认设备或数据类型，例如使用 device='cuda' 。