5c692f345e539ecf5d3a4b37d8150cea82ee41bf,pixyz/distributions/flow_distribution.py,TransformedDistribution,forward,#TransformedDistribution#Any#Any#Any#,122

Before Change

        z : torch.Tensor

        
        return self.flow.forward(x=x, y=y, compute_jacobian=compute_jacobian)

    def inverse(self, z, y=None):
        
        Backward (inverse) propagation of flow layers.

After Change

        z : torch.Tensor

        
        z = self.flow.forward(x=x, y=y, compute_jacobian=compute_jacobian)

        self.stored_x.clear()
        self.stored_x[hash(z)] = x
        return z

    def inverse(self, z, y=None):
        
        Backward (inverse) propagation of flow layers.

Instances