9a244189f1fe351a4fc4ce47e928bd3b3163fa2e,Tars/distributions/real_nvp.py,RealNVP,forward,#RealNVP#Any#Any#Any#,45

Before Change

        if jacobian is False:
            if inverse:
                x_1 = (x_1 - trans) * torch.exp(-scale)
            else:
                x_1 = x_1 * torch.exp(scale) + trans
            output = torch.cat((x_0, x_1), dim=-1)

        else:
            output = torch.sum(scale, dim=-1)

After Change

        if jacobian is False:
            for flow in _flows:
                x = flow(x, inverse=inverse)
            output = x

        else:
            logdet_jacobian = 0

Instances