def METHOD_NAME(self, x):
    """Updates the mean, var and count from a batch of samples."""
    batch_mean = np.mean(x, axis=0)
    batch_var = np.var(x, axis=0)
    batch_count = x.shape[0]
    self.update_from_moments(batch_mean, batch_var, batch_count)