RSSMPosterior

class torchrl.modules.RSSMPosterior(hidden_dim=200, state_dim=30, scale_lb=0.1)

RSSM 的後驗網路。

此網路將信念和相關的編碼觀察作為輸入。 它會傳回後驗的參數，以及根據此分配取樣的狀態。

參考資料：https://arxiv.org/abs/1811.04551

參數:

• hidden_dim (int, optional) – 線性網路中隱藏單元的數量。預設為 200。

• state_dim (int, optional) – 狀態的大小。預設為 30。

• scale_lb (float, optional) – 狀態分配的縮放比例下限。預設為 0.1。

forward(belief, obs_embedding)

定義每次呼叫時執行的計算。

應該由所有子類別覆寫。

注意

雖然 forward 傳遞的配方需要在這個函式中定義，但應該在之後呼叫 Module 實例，而不是呼叫此函式，因為前者會負責執行已註冊的掛鉤，而後者會默默地忽略它們。