Have a question? Connect with the community at the TensorFlow Forum Visit Forum

Module: tf_agents.utils.tensor_normalizer

Tensor statistics and normalization.


class EMATensorNormalizer: TensorNormalizer with exponential moving avg. mean and var estimates.

class StreamingTensorNormalizer: Normalizes mean & variance based on full history of tensor values.

class TensorNormalizer: Encapsulates tensor normalization and owns normalization variables.


parallel_variance_calculation(...): Calculate the sufficient statistics (average & second moment) of two sets.