Method mode

diff2flow/kl_autoencoder.py:82–83

(self)

            dim=dims)

    def mode(self):
        return self.mean
    

""" Resnet blocks """