tfp.experimental.stats.RunningMeanState

RunningMeanState(num_samples, mean)

num_samples

mean