VanillaMlpEncoder

source

VanillaMlpEncoder(
   observation_space: gym.Space, feature_dim: int = 64, hidden_dim: int = 64
)

---

Multi layer perceptron (MLP) for processing state-based inputs.

Args

• observation_space (gym.Space) : Observation space.
• feature_dim (int) : Number of features extracted.
• hidden_dim (int) : Number of hidden units in the hidden layer.

Returns

Mlp-based encoder instance.

Methods:

.forward

source

.forward(
   obs: th.Tensor
)

---

Forward method implementation.

Args

• obs (th.Tensor) : Observation tensor.

Returns

Encoded observation tensor.