コード例 #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)
コード例 #2
0
ファイル: test_univariate.py プロジェクト: zmoon111/zhusuan
    def test_batch_shape(self):
        def _distribution(param):
            return Binomial(param, 10)

        utils.test_batch_shape_1parameter(self,
                                          _distribution,
                                          np.ones,
                                          is_univariate=True)
コード例 #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)
コード例 #4
0
 def test_batch_shape(self):
     utils.test_batch_shape_1parameter(self,
                                       Dirichlet,
                                       np.zeros,
                                       is_univariate=False)
コード例 #5
0
 def test_batch_shape(self):
     utils.test_batch_shape_1parameter(self,
                                       OnehotCategorical,
                                       np.zeros,
                                       is_univariate=False)
コード例 #6
0
 def test_batch_shape(self):
     utils.test_batch_shape_1parameter(self,
                                       UnnormalizedMultinomial,
                                       np.zeros,
                                       is_univariate=False)
コード例 #7
0
ファイル: test_univariate.py プロジェクト: zmoon111/zhusuan
 def test_batch_shape(self):
     utils.test_batch_shape_1parameter(self,
                                       Poisson,
                                       np.ones,
                                       is_univariate=True)
コード例 #8
0
ファイル: test_univariate.py プロジェクト: zmoon111/zhusuan
 def test_batch_shape(self):
     utils.test_batch_shape_1parameter(self,
                                       Bernoulli,
                                       np.zeros,
                                       is_univariate=True)