使用 PyTorch 计算两个向量的余弦相似度

下面是使用 PyTorch 计算两个向量余弦相似度的 Python 代码：

import torch

# 定义两个向量
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])

# 计算余弦相似度
cos_sim = torch.dot(a, b) / (torch.norm(a) * torch.norm(b))

print(cos_sim.item()) # 打印余弦相似度的值

输出结果为：

0.9746316070556641

这个值表示向量'a' 和向量'b' 的余弦相似度为 0.9746，说明它们的方向非常相似。