def set_by_sample(self, train=True):
xp = self.xp
use_gpu = (xp == cuda.cupy)
for i in range(self.num_layers):
# h
mu, sigma = self.hmus[i], self.hsigmas[i]
e = np.random.normal(0., 1., self.z_size).astype(np.float32)
if use_gpu:
e = cuda.to_gpu(e)
self.decoder.set_h(i, self.get_zh(i)(mu + e * sigma))
# c
mu, sigma = self.cmus[i], self.csigmas[i]
e = np.random.normal(0., 1., self.z_size).astype(np.float32)
if use_gpu:
e = cuda.to_gpu(e)
self.decoder.set_c(i, self.get_zc(i)(mu + e * sigma))
评论列表
文章目录