lit_sb3.modules.feature_extractors.flatten

source module lit_sb3.modules.feature_extractors.flatten

Classes

• FlattenExtractor — Feature extract that flatten the input.

source class FlattenExtractor(observation_space: gym.Space)

Bases : BaseFeaturesExtractor

Feature extract that flatten the input.

Used as a placeholder when feature extraction is not needed.

Parameters

• observation_space : gym.Space — The observation space of the environment

Attributes

• features_dim : int — The number of features that the extractor outputs.

Methods

• forward

source method FlattenExtractor.forward(observations: torch.Tensor) → torch.Tensor