未初始化参数¶

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

一个未初始化的参数。

未初始化的参数是 torch.nn.Parameter 的一种特殊情况，其中数据的形状仍然未知。

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

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