pytorch 2范数归一化的表示

def l2_norm(input, axit=1):
    norm = torch.norm(input,2,axit,True)
    output = torch.div(input, norm)
    return output
# 这里假设input.shape = (2, 10) 最后得到的结果是每一行得到的每个值平方和为1

self.kernel.data.uniform_(-1, 1).renorm_(2, 1, 1e-5).mul_(1e5)
# 这行代码假设开始self.kernel.shape = (3 , 10)最终得到的每一列所有元素平方和为1

以上两种norm方式做个简单的笔记