Output of initializer.
| 49 | |
| 50 | |
| 51 | class InitializerOutput(NamedTuple): |
| 52 | """Output of initializer.""" |
| 53 | |
| 54 | # Gaussian base values. |
| 55 | gaussian_base_values: GaussianBaseValues |
| 56 | |
| 57 | # Feature input to the Gaussian predictor. |
| 58 | feature_input: torch.Tensor |
| 59 | |
| 60 | # Global scale to unscale output. |
| 61 | global_scale: torch.Tensor | None = None |
| 62 | |
| 63 | |
| 64 | class MultiLayerInitializer(nn.Module): |