def update(self, x):
        """
            Count x into historical average, updates running sums and count
            Accepts batched input
            Args:
                x: input tensor to be kept in record. 
        """
        self.count += float(np.prod(x.size()))
        self.running_sum += x.sum()
        self.running_sumsq = (x * x).sum()

            Count x into historical average, updates running sums and count
            Accepts batched input
            Args:
                x: input tensor to be kept in record. 
        
surreal/model/reward_filter.py