快捷方式

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

返回:

一个可调用对象 (Callable)

文档

访问全面的 PyTorch 开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源