快捷方式

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 開發者文件

檢視文件

教程

為初學者和高階開發者提供深入的教程

檢視教程

資源

查詢開發資源並讓您的問題得到解答

檢視資源