Example #1
0
    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)
Example #2
0
 def gaussian_encoder_with_gfeats(self, input_tensor, is_training):
     """
     For training use.
     """
     return architectures.make_gaussian_encoder(input_tensor,
                                                is_training=is_training)