未初始化参数

类torch.nn.parameter.UninitializedParameter(requires_grad=True, device=None, dtype=None)[源代码]

一个未初始化的参数。

未初始化的参数是torch.nn.Parameter的一种特殊情况，此时数据的形状仍不清楚。

与 torch.nn.Parameter 不同，未初始化的参数不包含任何数据，并且尝试访问某些属性（如形状）会抛出运行时错误。对未初始化参数可以执行的操作包括更改其数据类型、将其移动到不同的设备以及将其转换为常规 torch.nn.Parameter。

在构建时，可以使用例如 device='cuda' 来设置参数在实体化时的默认设备或数据类型。