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

Guess you like

Origin blog.csdn.net/cobracanary/article/details/120849992