addf94ed7dd37459b8b72964713f28328ae699b1,tests/test_distributions.py,DistributionsTestCase,test_dist_normal_batched_2,#DistributionsTestCase#,193

Before Change

        dist_batch_shape_correct = torch.Size([2, 3])
        dist_event_shape_correct = torch.Size()
        dist_sample_shape_correct = torch.Size([2, 3])
        dist_means_correct = [[0, 2, 0], [2, 0, 2]]
        dist_stddevs_correct = [[1, 3, 1], [3, 1, 3]]
        dist_log_probs_correct = [[-0.918939, -2.01755, -0.918939], [-2.01755, -0.918939, -2.01755]]

After Change

        dist_stddevs = util.to_numpy(dist.stddev)
        dist_stddevs_empirical = util.to_numpy(dist_empirical.stddev)
        dist_log_probs = util.to_numpy(dist.log_prob(dist_means_correct))
        dist_log_prob_shape = dist.log_prob(dist_means_correct).size()

        util.debug("dist_batch_shape", "dist_batch_shape_correct", "dist_event_shape", "dist_event_shape_correct", "dist_sample_shape", "dist_sample_shape_correct", "dist_log_prob_shape", "dist_log_prob_shape_correct", "dist_means", "dist_means_empirical", "dist_means_correct", "dist_stddevs", "dist_stddevs_empirical", "dist_stddevs_correct", "dist_log_probs", "dist_log_probs_correct")

        self.assertEqual(dist_batch_shape, dist_batch_shape_correct)

Instances