torch归一化

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

实例:

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

你可能感兴趣的:(graph,ML&&DL,pytorch)