评价此页

UninitializedBuffer#

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

一个未初始化的缓冲区。

Uninitialized Buffer 是 torch.Tensor 的一个特例,其中数据的形状仍然未知。

torch.Tensor 不同,未初始化的参数不包含任何数据,尝试访问其形状等属性会引发运行时错误。对未初始化参数可以进行的操作只有更改其数据类型、将其移动到不同的设备以及将其转换为常规的 torch.Tensor

可以使用 `device='cuda'` 等方式在构造期间设置当缓冲区具体化时使用的默认设备或数据类型。