UninitializedBuffer#
- class torch.nn.parameter.UninitializedBuffer(requires_grad=False, device=None, dtype=None, persistent=True)[source]#
一个未初始化的 buffer。
Uninitialized Buffer 是
torch.Tensor
的一个特殊情况,其数据形状仍未知。与
torch.Tensor
不同,未初始化的参数不包含数据,尝试访问其某些属性(例如形状)将引发运行时错误。可以在未初始化的参数上执行的唯一操作是更改其数据类型、将其移动到不同的设备以及将其转换为常规的torch.Tensor
。当 buffer 被具体化时使用的默认设备或 dtype 可以在构造期间设置,例如使用
device='cuda'
。