def test_sample(self): def _distribution(param): return Multinomial(param, n_experiments=10) utils.test_1parameter_sample_shape_one_rank_less( self, _distribution, np.zeros) dist = Multinomial(np.ones([2, 2]), n_experiments=None) with self.assertRaisesRegexp(ValueError, "Cannot sample when `n_experiments`"): dist.sample()
def test_sample_shape(self): def _distribution(param): return Multinomial(param, 10) utils.test_1parameter_sample_shape_one_rank_less( self, _distribution, np.zeros)
def test_sample_shape(self): def _proxy_distribution(logits): return Concrete(1., logits) utils.test_1parameter_sample_shape_one_rank_less( self, _proxy_distribution, np.zeros)
def test_sample_shape(self): utils.test_1parameter_sample_shape_one_rank_less( self, Dirichlet, np.zeros)
def test_sample_shape(self): utils.test_1parameter_sample_shape_one_rank_less( self, OnehotCategorical, np.zeros)