做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]])