Method forward

diff2flow/kl_autoencoder.py:521–529

(self, input, sample_posterior=True)

        return dec

    def forward(self, input, sample_posterior=True):
        posterior = self.encode(input, return_posterior=True)
        if sample_posterior:
            z = posterior.sample()
        else:
            z = posterior.mode()
        z = (z + self.shift) * self.scale
        dec = self.decode(z)
        return dec, posterior


if __name__ == "__main__":