def _count_samples()

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