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)

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

應由所有子類重寫。

注意

儘管前向傳播的實現需要在本函式內定義，但後續應呼叫 Module 例項而不是直接呼叫此函式，因為前者會處理已註冊的鉤子，而後者會默默忽略它們。