快捷方式

biased_softplus

class torchrl.modules.utils.biased_softplus(bias: float, min_val: float = 0.01)[来源]

一个有偏置的 softplus 模块。

偏置指示当零张量通过变换时要返回的值。

参数:
  • bias (标量) – softplus 变换的“偏置”。如果 bias=1.0,则将计算 _bias 偏移,使得 softplus(0.0 + _bias) = bias。

  • min_val (标量) – 变换的最小值。默认值:0.1

forward(x: Tensor) Tensor[来源]

定义每次调用时执行的计算。

应由所有子类重写。

注意

尽管 forward 传递的配方需要在该函数内定义,但之后应调用 Module 实例而不是此函数,因为前者负责运行注册的钩子,而后者会静默地忽略它们。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源