Exemplo n.º 1
0
    def test_batch_shape(self):
        def _distribution(param):
            return Multinomial(param, 10)

        utils.test_batch_shape_1parameter(self,
                                          _distribution,
                                          np.zeros,
                                          is_univariate=False)
Exemplo n.º 2
0
    def test_batch_shape(self):
        def _distribution(param):
            return Binomial(param, 10)

        utils.test_batch_shape_1parameter(self,
                                          _distribution,
                                          np.ones,
                                          is_univariate=True)
Exemplo n.º 3
0
    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)
Exemplo n.º 4
0
 def test_batch_shape(self):
     utils.test_batch_shape_1parameter(self,
                                       Dirichlet,
                                       np.zeros,
                                       is_univariate=False)
Exemplo n.º 5
0
 def test_batch_shape(self):
     utils.test_batch_shape_1parameter(self,
                                       OnehotCategorical,
                                       np.zeros,
                                       is_univariate=False)
Exemplo n.º 6
0
 def test_batch_shape(self):
     utils.test_batch_shape_1parameter(self,
                                       UnnormalizedMultinomial,
                                       np.zeros,
                                       is_univariate=False)
Exemplo n.º 7
0
 def test_batch_shape(self):
     utils.test_batch_shape_1parameter(self,
                                       Poisson,
                                       np.ones,
                                       is_univariate=True)
Exemplo n.º 8
0
 def test_batch_shape(self):
     utils.test_batch_shape_1parameter(self,
                                       Bernoulli,
                                       np.zeros,
                                       is_univariate=True)