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