def encode_x_a(self, x, test=False):
x = self.to_variable(x)
mean, ln_var = self.q_a_x(x, test=test)
return F.gaussian(mean, ln_var)
微信
公众号
扫码关注公众号