DreamerActor

class torchrl.modules.DreamerActor(out_features, depth=4, num_cells=200, activation_class=<class 'torch.nn.modules.activation.ELU'>, std_bias=5.0, std_min_val=0.0001)

Dreamer actor 網路。

此網路用於根據當前時間步的隨機狀態和確定性信念預測動作分佈。它輸出動作分佈的均值和尺度。

參考：https://arxiv.org/abs/1912.01603

引數：

• out_features (int) – 輸出特徵的數量。

• depth (int, optional) – 隱藏層的數量。預設為 4。

• num_cells (int, optional) – 每層隱藏單元的數量。預設為 200。

• activation_class (nn.Module, optional) – 啟用類。預設為 nn.ELU。

• std_bias (float, optional) – softplus 變換的偏差。預設為 5.0。

• std_min_val (float, optional) – 標準差的最小值。預設為 1e-4。

forward(state, belief)

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

應被所有子類覆蓋。

注意

儘管 forward 通行的實現需要在該函式中定義，但後續應該呼叫 Module 例項而不是此函式本身，因為前者負責執行註冊的鉤子，而後者則會默默忽略它們。