ray.rllib.utils.exploration.random_encoder.RE3.postprocess_trajectory

RE3.postprocess_trajectory(policy, sample_batch, tf_sess=None)

Calculate states’ latent representations/embeddings.

Embeddings are added to the SampleBatch object such that it doesn’t need to be calculated during each training step.