neuralcompression/models/_mean_scale_hyperprior_autoencoder.py [101:114]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            minimum,
            maximum,
            steps,
        )

    def forward(
        self,
        x: Tensor,
    ) -> Tuple[Tensor, Tuple[Tensor, Tensor]]:
        y = self.encoder(x)

        z = self.hyper_encoder(y)

        z_hat, z_scores = self.bottleneck(z)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



neuralcompression/models/_scale_hyperprior_autoencoder.py [49:62]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            minimum,
            maximum,
            steps,
        )

    def forward(
        self,
        x: Tensor,
    ) -> Tuple[Tensor, Tuple[Tensor, Tensor]]:
        y = self.encoder(x)

        z = self.hyper_encoder(y)

        z_hat, z_scores = self.bottleneck(z)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -