コード例 #1
0
ファイル: test_distribution.py プロジェクト: ypxie/chainerrl 
 def setUp(self):
     self.logits = np.random.rand(self.batch_size, self.n)
     if self.wrap_by_variable:
         self.distrib = distribution.SoftmaxDistribution(
             chainer.Variable(self.logits))
     else:
         self.distrib = distribution.SoftmaxDistribution(self.logits)
コード例 #2
0
 def __call__(self, x):
     h = self.model(x)
     return distribution.SoftmaxDistribution(
         h, beta=self.beta, min_prob=self.min_prob)
コード例 #3
0
 def __call__(self, x, test=False):
     h = self.model(x, test=test)
     return distribution.SoftmaxDistribution(h,
                                             beta=self.beta,
                                             min_prob=self.min_prob)