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)