未初始化的缓冲区
- 类torch.nn.parameter.UninitializedBuffer(requires_grad=False, device=None, dtype=None, persistent=True)[源代码]
-
一个未初始化的缓冲区。
未初始化的缓冲区是
torch.Tensor
的一种特殊情况,此时数据的形状仍不清楚。与
torch.Tensor
不同,未初始化的参数不包含任何数据。尝试访问其属性(如形状)将会抛出运行时错误。对未初始化参数可以执行的操作仅限于更改其数据类型、将其移动到不同的设备以及将其转换为常规torch.Tensor
。可以在构造时使用例如
device='cuda'
来设置缓冲区实例化的默认设备或数据类型。