Function _sum_ft

lib/nn/modules/batchnorm.py:24–26

sum over the first and last dimention

(tensor)

def _sum_ft(tensor):
    """sum over the first and last dimention"""
    return tensor.sum(dim=0).sum(dim=-1)


def _unsqueeze_ft(tensor):