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

Before Change

        return scale, trans

    def forward(self, x, inverse=False, jacobian=False):
        x_0 = x[:, :self.masked_features]
        x_1 = x[:, self.masked_features:]
        scale, trans = self._scale_translation(x_0)

        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

    def forward(self, x, inverse=False, jacobian=False):
        if inverse is False:
            _flows = self.flows
        else:
            _flows = self.flows[::-1]

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

        else:
            logdet_jacobian = 0
            for flow in _flows:
                x, _logdet_jacobian = flow(x,
                                           inverse=inverse,
                                           jacobian=jacobian)
                logdet_jacobian += _logdet_jacobian
            output = logdet_jacobian

        return output

Instances