intermediate#

Source code: tianshou/highlevel/module/intermediate.py

class IntermediateModule(module: Module, output_dim: int)[source]#

Bases: object

Container for a module which computes an intermediate representation (with a known dimension).

module: Module#
output_dim: int#
get_module_with_vector_output() → ModuleWithVectorOutput[source]#
class IntermediateModuleFactory[source]#

Bases: ToStringMixin, ModuleFactory, ABC

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]#