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 添加自定义映射。

返回:

一个可调用对象