赞
踩

以上就是余弦相似度的所有内容
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))
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。