快捷方式

未初始化缓冲区

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

一个未初始化的缓冲区。

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

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

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源