未初始化缓冲区¶
- class torch.nn.parameter.UninitializedBuffer(requires_grad=False, device=None, dtype=None)[源代码]¶
一个未初始化的缓冲区。
未初始化的缓冲区是
torch.Tensor
的一种特殊情况,其中数据的形状仍然未知。与
torch.Tensor
不同,未初始化的参数不持有数据,尝试访问某些属性(如它们的形状)将抛出运行时错误。唯一可以对未初始化参数执行的操作是更改其数据类型、将其移动到不同的设备以及将其转换为常规的torch.Tensor
。在缓冲区具体化时使用的默认设备或dtype可以在构造期间通过例如
device='cuda'
来设置。