def compute_mean_and_var(self, x):
# mean = self.mean_layer(x)
mean = F.tanh(self.mean_layer(x)) * 2.0
var = F.softplus(self.var_layer(x))
return mean, var
微信
公众号
扫码关注公众号