ObsEncoder

class torchrl.modules.ObsEncoder(channels=32, num_layers=4, depth=None)

觀測編碼器網路。

接受畫素觀測並將其編碼到潛在空間。

參考: https://arxiv.org/abs/1803.10122

引數:

• channels (int, 可選) – 第一層中的隱藏單元數量。預設為 32。

• num_layers (int, 可選) – 網路的深度。預設為 4。

forward(observation)

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

應由所有子類重寫。

注意

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