mappings

class torchrl.modules.utils.mappings(key: str)

给定一个输入字符串，返回一个满射函数 f(x): R -> R^+。

参数:

key (str) – 以下之一："softplus", "exp", "relu", "expln", "biased_softplus" 或 "none" (无映射)。

注意

如果 key 以 “biased_softplus” 开头，则需要采用以下形式：`"biased_softplus_{bias}"`，其中 `bias` 可以转换为浮点数，用于对 softplus 函数进行偏置。或者，可以使用 `"biased_softplus_{bias}_{min_val}"` 语法。在这种情况下，额外的 `min_val` 项是一个浮点数，用于编码 softplus 变换的最小值。实际上，使用的公式是 softplus(x + bias) + min_val，其中 bias 和 min_val 是经过计算的值，以满足上述条件。

注意

可以通过 tensordict.nn.add_custom_mapping 添加自定义映射。

返回:

一个可调用对象