欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

求两组数据的两两之间的欧氏距离的方法

程序员文章站 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]])