Pytorch框架中余弦相似度(Cosine similarity)、欧氏距离(Euclidean distance)源码解析
一、矩阵操作用于计算余弦相似度余弦相似度:我们知道,分子是矩阵的乘法,分母是两个标量的乘积。分母好办,关键是如何在计算分子?很简单,我们可以将公式变变形:那么我们只需在矩阵乘法前,使其归一化,乘法之后就是余弦相似度了,来看一下代码(参考:https://zhuanlan.zhihu.com/p/383675457)importtorch##计算两个特征的余弦相似度defnormalize(x,ax