NoisyLazyLinear

class torchrl.modules.NoisyLazyLinear(out_features: int, bias: bool = True, device: DEVICE_TYPING | None = None, dtype: torch.dtype | None = None, std_init: float = 0.1)

帶噪聲的懶惰線性層。

此類使 Noisy Linear 層“懶惰”，即在初始化時無需傳遞 in_feature 引數（但在呼叫該層第一次之後會自動推斷）。

有關帶噪聲層更多背景資訊，請參閱 NoisyLinear 類。

引數:

• out_features (int) – 輸出特徵維度

• bias (bool, optional) – 如果為 True，則會在矩陣乘法 Ax + b 中新增一個偏置項。預設為 True。

• device (DEVICE_TYPING, optional) – 層的裝置。預設為 "cpu"。

• dtype (torch.dtype, optional) – 引數的資料型別。預設為 PyTorch 的預設資料型別。

• std_init (scalar) – 最佳化前的��數標準差的初始值。預設為 0.1

initialize_parameters(input: Tensor) → None

Initialize parameters according to the input batch properties.

This adds an interface to isolate parameter initialization from the forward pass when doing parameter shape inference.