def z_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. 
        """
        if x is None: return

        # only called in learner, so we can assume it has the correct type
        if len(x.size()) == 3: x = x.view(-1, self.in_size)
        self.running_sum += torch.sum(x, dim=0)
        self.running_sumsq += torch.sum(x * x, dim=0)
        self.count += float(len(x))

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