快捷方式

biased_softplus

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

带偏置的 softplus 模块。

该 bias 指示了当输入为零张量时要返回的值。

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

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

forward(x: Tensor) Tensor[source]

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

所有子类都应重写此方法。

注意

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

文档

访问全面的 PyTorch 开发者文档

查看文档

教程

为初学者和高级开发者提供深入的教程

查看教程

资源

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

查看资源