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)
def __call__(self, x): h = self.model(x) return distribution.SoftmaxDistribution( h, beta=self.beta, min_prob=self.min_prob)
def __call__(self, x, test=False): h = self.model(x, test=test) return distribution.SoftmaxDistribution(h, beta=self.beta, min_prob=self.min_prob)