8f200a4af5f4708e82dd7d62df8620268a97bebd,pyro/distributions/mixture.py,MaskedMixture,rsample,#MaskedMixture#Any#,109

Before Change

    def rsample(self, sample_shape=torch.Size()):
        mask = self.mask.expand(sample_shape + self.batch_shape) if sample_shape else self.mask
        result = self.component0.rsample(sample_shape)
        result[mask] = self.component1.rsample(sample_shape)[mask]
        return result

    def log_prob(self, value):

After Change

    def rsample(self, sample_shape=torch.Size()):
        mask = self.mask.reshape(self.mask.shape + (1,) * self.event_dim)
        mask = mask.expand(sample_shape + self.shape())
        result = torch.where(mask,
                             self.component1.rsample(sample_shape),
                             self.component0.rsample(sample_shape))
        return result

Instances