Module: tf_agents.utils.tensor_normalizer

Tensor statistics and normalization.

Classes

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.

kahan_summation(...): Calculate stable acculated sum using compensation for low-bits.

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

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2024-04-26 UTC.