Method forward

sat/sgm/modules/autoencoding/magvit2_pytorch.py:469–473

(self, x: Tensor, *, cond: Optional[Tensor] = None)

    @beartype
    def forward(self, x: Tensor, *, cond: Optional[Tensor] = None):
        maybe_cond_kwargs = dict(cond=cond) if exists(cond) else dict()

        x = self.norm(x, **maybe_cond_kwargs)
        return self.net(x)


# discriminator with anti-aliased downsampling (blurpool Zhang et al.)