ObsEncoder

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

觀測編碼器網路。

將畫素觀測編碼到潛在空間。

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

引數:

• channels (int, optional) – 第一個隱藏單元的數量。預設為 32。

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

forward(observation)

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

所有子類都應重寫此方法。

注意

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