快捷方式

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)[source]

带噪声的惰性线性层。

此类使 Noisy Linear 层具有惰性,即在初始化时无需传入 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[source]

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

这提供了一个接口,用于在进行参数形状推断时,将参数初始化与前向传播过程隔离开来。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题解答

查看资源