kd.nn.FlatAutoencoder#

class kauldron.modules.FlatAutoencoder( encoder: flax.linen.module.Module, decoder: flax.linen.module.Module, inputs: typing.Annotated[typing.Any, <object object at 0x7824c478ba80>] = '__KEY_REQUIRED__', parent: flax.linen.module.Module | flax.core.scope.Scope | flax.linen.module._Sentinel | None = <flax.linen.module._Sentinel object>, name: str | None = None, )[source]

Bases: flax.linen.module.Module

Very simple auto-encoder class to showcase using keys and submodules.

encoder: nn.Module

decoder: nn.Module

inputs: kontext.Key = '__KEY_REQUIRED__'

name: str | None = None

parent: flax.linen.module.Module | flax.core.scope.Scope | flax.linen.module._Sentinel | None = None

scope: Scope | None = None