torch归一化

做GNN的人会常常需要将矩阵进行行归一化，之前一直是自己手动实现归一化，现在发现一个很好用的API：torch.nn.functional.normalize()
看看官方文档：

实例：

>>> x
tensor([[ 1,  2, 34],
        [ 4,  5,  6]])

>>> F.normalize(x.float(),p=1,dim=1)
tensor([[0.0270, 0.0541, 0.9189],
        [0.2667, 0.3333, 0.4000]])

标签：normalize,torch,0.2667,float,归一化,tensor
来源： https://blog.csdn.net/cobracanary/article/details/120849992