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 例項而不是此函式，因為前者會處理註冊的鉤子，而後者則會靜默忽略它們。