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

求两组数据的两两之间的欧氏距离的方法如下（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]])