import tensorflow as tf


  def inject_latent(self, layer, features, filters):
    """Inject a VAE-style latent."""
    # Latent for stochastic model
    input_frames = tf.to_float(features["inputs_raw"])
    target_frames = tf.to_float(features["targets_raw"])
    full_video = tf.concat([input_frames, target_frames], axis=1)
    latent_mean, latent_std = self.construct_latent_tower(
        full_video, time_axis=1)
    latent = common_video.get_gaussian_tensor(latent_mean, latent_std)
    latent = tf.layers.flatten(latent)
    latent = tf.expand_dims(latent, axis=1)
