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 pass 的实现需要在该函数中定义，但应该调用 Module 实例而不是直接调用此函数，因为前者负责运行注册的钩子，而后者则会静默忽略它们。