pytorch 余弦相似性

PyTorch中的余弦相似性可以使用torch.nn.functional中的cosine_similarity函数来计算。该函数接受两个张量作为输入，并返回它们之间的余弦相似度。

示例代码如下：

import torch.nn.functional as F

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

# 计算余弦相似度
similarity = F.cosine_similarity(x, y)
print(similarity)

输出结果为：

tensor(0.9746)