TassaCnnEncoder

source

TassaCnnEncoder(
   observation_space: gym.Space, feature_dim: int = 50
)

---

Convolutional neural network (CNN)-based encoder for processing image-based observations. Proposed by Tassa Y, Doron Y, Muldal A, et al. Deepmind control suite[J]. arXiv preprint arXiv:1801.00690, 2018. Target task: DeepMind Control Suite.

Args

• observation_space (gym.Space) : Observation space.
• feature_dim (int) : Number of features extracted by the encoder.

Returns

CNN-based encoder instance.

Methods:

.forward

source

.forward(
   obs: th.Tensor
)

---

Forward method implementation.

Args

• obs (th.Tensor) : Observation tensor.

Returns

Encoded observation tensor.