欧式距离相似度 python
时间: 2023-11-01 18:57:39 浏览: 82
算法:欧式距离算法,python源码
欧式距离是用来衡量两个向量之间的相似性或差异程度的一种度量方法。在Python中,可以使用torch.PairwiseDistance函数来计算欧式距离。
首先,导入相关的库和模块:
```python
import torch
import torch.nn as nn
```
然后,创建需要比较的两个向量a和b:
```python
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
```
接下来,使用torch.PairwiseDistance函数计算欧式距离:
```python
euclidean_distance = nn.PairwiseDistance()
distance = euclidean_distance(a, b)
print(distance.item())
```
以上代码将输出欧式距离的值。
阅读全文