in inplace_abn/functions.py [0:0]
def _count_samples(x): count = x.size(0) for i in range(2, x.ndimension()): count *= x.size(i) return count