1d version of evo_norms about torch-toolbox HOT 2 CLOSED

def instance_std_1d(x, eps=1e-5):
    var = torch.var(x, dim=(2, ), keepdim=True)
    std = torch.sqrt(var + eps)
    return std

def group_std_1d(x: torch.Tensor, groups=32, eps=1e-5):
    n, c, l = x.size()
    x = torch.reshape(x, (n, groups, c // groups, l))
    var = torch.var(x, dim=(2, 3), keepdim=True)
    std = torch.sqrt(var + eps)
    return torch.reshape(std, (n, c, h, w))

var = torch.var(x, dim=(0, 2), keepdim=True)