GAN_MNIST_jax module

class GAN_MNIST_jax.Discriminator_MNIST_cond(parent: Union[flax.linen.module.Module, flax.core.scope.Scope, flax.linen.module._Sentinel, NoneType] = <flax.linen.module._Sentinel object at 0x7f0f496f5960>, name: Optional[str] = None)

Bases: Module

name: str | None = None

parent: Module | Scope | _Sentinel | None = None

scope: Scope | None = None

class GAN_MNIST_jax.Generator_MNIST_cond(latent_dim: int = 118, num_classes: int = 10, parent: Union[flax.linen.module.Module, flax.core.scope.Scope, flax.linen.module._Sentinel, NoneType] = <flax.linen.module._Sentinel object at 0x7f0f496f5960>, name: Optional[str] = None)

Bases: Module

latent_dim: int = 118

name: str | None = None

num_classes: int = 10

parent: Module | Scope | _Sentinel | None = None

scope: Scope | None = None