def setUp(self): self.values = np.random.rand(self.batch_size, self.n) if self.wrap_by_variable: self.distrib = distribution.MellowmaxDistribution( chainer.Variable(self.values)) else: self.distrib = distribution.MellowmaxDistribution(self.values)
def __call__(self, x): h = self.model(x) return distribution.MellowmaxDistribution(h, omega=self.omega)
def __call__(self, x, test=False): h = self.model(x, test=test) return distribution.MellowmaxDistribution(h, omega=self.omega)