def prepare(self): self._mu_encoder = Dense(self.size, 'linear', init=GaussianInitializer(), random_bias=True).init(self.input_dim) self._log_sigma_encoder = Dense(self.size, 'linear', init=GaussianInitializer(), random_bias=True).init(self.input_dim) self.register_inner_layers(self._mu_encoder, self._log_sigma_encoder)