求两组数据的两两之间的欧氏距离的方法
程序员文章站
2022-06-03 17:45:14
...
求两组数据的两两之间的欧氏距离的方法如下(tensor):
>>>import torch
>>> a=torch.Tensor([[1,1,3],[2,2,3]])
>>> a
tensor([[1., 1., 3.],
[2., 2., 3.]])
>>> b=torch.Tensor([4,1,3])
>>> b
tensor([4., 1., 3.])
>>> dist = torch.sqrt(torch.sum((a[:,None,:] - b) ** 2, dim=2))
>>> dist
tensor([[3.0000],
[2.2361]])
上一篇: 兄弟做不成了