Reinforcement Learning, Gaming, Deep Learning.
batch_mean = torch.mean(x, dim=0) batch_var = torch.var(x, dim=0) batch_count = x.shape[0] self.update_from_moments(batch_mean, batch_var, batch_count) def update_from_moments(self, batch_mean, batch_var, batch_count): self.mean, self.var, self.count = update_mean_var_count_from_moments( self.mean,