def gaussian_encoder(self, input_tensor, is_training): """Applies the Gaussian encoder to images. Args: input_tensor: Tensor with the observations to be encoded. is_training: Boolean indicating whether in training mode. Returns: Tuple of tensors with the mean and log variance of the Gaussian encoder. """ return architectures.make_gaussian_encoder(input_tensor, is_training=is_training)
def gaussian_encoder_with_gfeats(self, input_tensor, is_training): """ For training use. """ return architectures.make_gaussian_encoder(input_tensor, is_training=is_training)