def METHOD_NAME(self, input: Tensor) -> bool:
    """Tracks statistics of a micro-batch."""
    # Dimensions except channel. For example, (0, 2, 3) is for BatchNorm2d.
    dim = [0]
    dim.extend(range(2, input.dim()))
    with torch.no_grad():
        self.sum += input.sum(dim)
        self.sum_squares += (input ** 2).sum(dim)
    size = input.size().numel() // input.size(1)
    self.counter += size
    self.tracked += 1
    return self.tracked == self.chunks