welcome to my blog
问题描述: 想对向量归一化, 或者说 让向量除以模长
解决方法, 看示例
import torch
import torch.nn.functional as F
a = torch.arange(9, dtype= torch.float)
a = a.reshape((3,3))
print(a)
'''
tensor([[0., 1., 2.],
[3., 4., 5.],
[6., 7., 8.]])
'''
# 对二维数组按行归一化
# p=2表示二范式, dim=1表示按行归一化
b = F.normalize(a, p=2, dim=1)
print(b)
'''
tensor([[0.0000, 0.4472, 0.8944],
[0.4243, 0.5657, 0.7071],
[0.4915, 0.5735, 0.6554]])
'''