Function normalize_tensor

vtp/utils/lpips.py:169–171

(x, eps=1e-10)

def normalize_tensor(x, eps=1e-10):
    norm_factor = torch.sqrt(torch.sum(x**2, dim=1, keepdim=True))
    return x / (norm_factor + eps)


def spatial_average(x, keepdim=True):