这个屌丝很懒，什么也没留下！

pytorch 计算cosine余弦相似度_pytorch f.cosine_similarity计算样本对cosine相似度

作者：编程变革者 | 2024-01-28 23:37:52

pytorch f.cosine_similarity计算样本对cosine相似度

import torch.nn.functional as F
F.cosine_similarity(a,b,dim=1)
1
2

沿着dim维度对a,b两个tensor计算余弦相似度。由于dim属性的存在，使得a,b两个tensor可以为任意维。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/article/detail/42254