intermediate#
Source code: tianshou/highlevel/module/intermediate.py
- class IntermediateModule(module: Module, output_dim: int)[source]#
Container for a module which computes an intermediate representation (with a known dimension).
- module: Module#
- output_dim: int#
- class IntermediateModuleFactory[source]#
Factory for the generation of a module which computes an intermediate representation.
- abstract create_intermediate_module(envs: Environments, device: str | device) IntermediateModule[source]#
- create_module(envs: Environments, device: str | device) Module[source]#