paddle计算两个向量相似度
时间: 2023-12-02 07:04:22 浏览: 272
PaddleDetection源码,PaddleDetection-master
你可以使用 PaddlePaddle 中的 `paddle.similarity_cosine` 函数计算两个向量的余弦相似度。
示例代码如下:
```python
import paddle
# 定义两个向量
vector1 = paddle.to_tensor([1, 2, 3])
vector2 = paddle.to_tensor([4, 5, 6])
# 计算两个向量的余弦相似度
similarity = paddle.similarity_cosine(vector1, vector2)
print(similarity.numpy()) # 输出结果为:0.9746318
```
其中,`similarity_cosine` 函数会接受两个形状相同的向量作为输入,并返回它们之间的余弦相似度。
阅读全文