余弦相似度，F.cosine_similarity()详解

以上就是余弦相似度的所有内容

import torch
import torch.nn.functional as F



##特征向量a
a = torch.tensor([[1, 2], [3, 4]], dtype=torch.float32)

##特征向量b
b = torch.tensor([[5, 6], [7, 8]], dtype=torch.float32)

print(F.cosine_similarity(a, b, dim=-1))