NoisyLazyLinear

class torchrl.modules.NoisyLazyLinear(out_features: int, bias: bool = True, device: Optional[Union[device, str, int]] = None, dtype: Optional[dtype] = None, std_init: float = 0.1)

噪声延迟线性层。

此类使噪声线性层延迟，即在初始化时不需要传递 in_feature 参数（但在第一次调用该层后会推断出来）。

有关噪声层的更多背景信息，请参阅 NoisyLinear 类。

参数：

• out_features (int) – 输出特征维度

• bias (bool, 可选) – 如果为 True，则将向矩阵乘法添加偏差项：Ax + b。默认为 True。

• device (DEVICE_TYPING, 可选) – 层的设备。默认为 "cpu"。

• dtype (torch.dtype, 可选) – 参数的数据类型。默认为 PyTorch 的默认数据类型。

• std_init (标量) – 优化前高斯标准偏差的初始值。默认为 0.1

initialize_parameters(input: Tensor) → None

根据输入批次属性初始化参数。

这添加了一个接口，用于在进行参数形状推断时将参数初始化与前向传递隔离开来。