未初始化的参数¶
- class torch.nn.parameter.UninitializedParameter(requires_grad=True, device=None, dtype=None)[source]¶
一个未初始化的参数。
未初始化的参数是
torch.nn.Parameter
的一种特殊情况,其中数据的形状仍然未知。与
torch.nn.Parameter
不同,未初始化的参数不包含任何数据,并且尝试访问某些属性(例如它们的形状)将引发运行时错误。可以对未初始化的参数执行的唯一操作是更改其数据类型,将其移动到不同的设备,并将其转换为常规的torch.nn.Parameter
。在参数具体化时要使用的默认设备或数据类型可以在构造期间设置,例如
device='cuda'
。