intermediate#


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