5184f1fc798a7728dd574ae9a351e16869d9ee7b,pyro/contrib/timeseries/lgssm.py,GenericLGSSM,_get_trans_dist,#GenericLGSSM#,54

Before Change

    def _get_trans_dist(self):
        loc = self.obs_matrix.new_zeros(self.state_dim)
        eye = torch.eye(self.state_dim, device=loc.device, dtype=loc.dtype)
        return MultivariateNormal(loc, self.log_trans_noise_scale_sq.exp() * eye)

    def _get_dist(self):
        

After Change

    def _get_obs_dist(self):
        return dist.Normal(self.obs_loc, self.obs_noise_scale).to_event(1)

    def _get_trans_dist(self):
        loc = self.obs_matrix.new_zeros(self.state_dim)
        return MultivariateNormal(loc, self.trans_noise_scale_sq.diag_embed())

    def _get_dist(self):
        

Instances