741108695943be12c93ff96a3cb964ddcbeb1563,pyro/infer/reparam/neutra.py,NeuTraReparam,__call__,#NeuTraReparam#Any#Any#Any#,58

Before Change

            // Sample a shared latent.
            // TODO(fehiepsi) Consider adding a method to extract transform from an Auto*Normal(posterior).
            posterior = self.guide.get_posterior()
            if not isinstance(posterior, dist.TransformedDistribution):
                raise ValueError("NeuTraReparam only supports guides whose posteriors are "
                                 "TransformedDistributions but got a posterior of type {}"
                                 .format(type(posterior)))
            self.transform = dist.transforms.ComposeTransform(posterior.transforms)
            z_unconstrained = pyro.sample("{}_shared_latent".format(name),
                                          posterior.base_dist.mask(False))

After Change

        log_density = 0.
        if not self.x_unconstrained:  // On first sample site.
            // Sample a shared latent.
            try:
                self.transform = self.guide.get_transform()
            except (NotImplementedError, TypeError):
                raise ValueError("NeuTraReparam only supports guides that implement "
                                 "`get_transform` method that does not depend on the "
                                 "model"s `*args, **kwargs`")

            z_unconstrained = pyro.sample("{}_shared_latent".format(name),
                                          self.guide.get_base_dist().mask(False))

            // Differentiably transform.

Instances