def _init_weight(self): init.xavier_normal(self._enc_mu.weight) init.xavier_normal(self._enc_log_sigma.weight)