当前位置:   article > 正文

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

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))
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/article/detail/42221
推荐阅读
相关标签
  

闽ICP备14008679号