Method mode

models/flowsep/diffusers/models/vae.py:440–441

(self)

        return 0.5 * torch.sum(logtwopi + self.logvar + torch.pow(sample - self.mean, 2) / self.var, dim=dims)

    def mode(self):
        return self.mean