快捷方式

biased_softplus

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

一个带偏差的 Softplus 模块。

偏差表示当通过转换传递零张量时要返回的值。

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

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

forward(x: Tensor) Tensor[source]

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

应由所有子类覆盖。

注意

虽然前向传递的配方需要在此函数中定义,但应随后调用Module实例,而不是此函数,因为前者负责运行注册的钩子,而后者则会静默忽略它们。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

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

查看资源