| 20 | @si_module |
| 21 | class GaussianMixtureIOLayer(nn.Module): |
| 22 | class Config: |
| 23 | latent_dim: int |
| 24 | dim: int |
| 25 | num_components: int |
| 26 | |
| 27 | def __init__(self, c: Config): |
| 28 | super().__init__() |
nothing calls this directly
no outgoing calls
no test coverage detected