def test_batch_shape(self): def _distribution(param): return Multinomial(param, 10) utils.test_batch_shape_1parameter(self, _distribution, np.zeros, is_univariate=False)
def test_batch_shape(self): def _distribution(param): return Binomial(param, 10) utils.test_batch_shape_1parameter(self, _distribution, np.ones, is_univariate=True)
def test_batch_shape(self): def _proxy_distribution(logits): return Concrete(1., logits) utils.test_batch_shape_1parameter(self, _proxy_distribution, np.zeros, is_univariate=False)
def test_batch_shape(self): utils.test_batch_shape_1parameter(self, Dirichlet, np.zeros, is_univariate=False)
def test_batch_shape(self): utils.test_batch_shape_1parameter(self, OnehotCategorical, np.zeros, is_univariate=False)
def test_batch_shape(self): utils.test_batch_shape_1parameter(self, UnnormalizedMultinomial, np.zeros, is_univariate=False)
def test_batch_shape(self): utils.test_batch_shape_1parameter(self, Poisson, np.ones, is_univariate=True)
def test_batch_shape(self): utils.test_batch_shape_1parameter(self, Bernoulli, np.zeros, is_univariate=True)