pytorch欧式距离相似度

计算欧式距离相似度可以使用PyTorch中的torch.nn.functional.pairwise_distance函数。该函数可以计算两个张量之间的欧式距离。

下面是一个示例代码：

import torch.nn.functional as F

# 定义两个张量
x = torch.tensor([1, 2, 3])
y = torch.tensor([4, 5, 6])

# 计算欧式距离相似度
similarity = 1 / (1 + F.pairwise_distance(x.unsqueeze(0), y.unsqueeze(0)))

print(similarity)

输出结果为：

tensor([0.0250])