pytorch计算一个矩阵行向量之间的相似度

前言

• 本文代码可在我的Github 中查看

直接上代码：

import torch as t
a = t.randn(2,4) # 随机初始化一个矩阵
print(a)

# step 1. 计算行向量的长度
len_a = t.sqrt(t.sum(a**2,dim=-1))
print(len_a)

b = len_a.unsqueeze(1).expand(-1,2)
c = len_a.expand(2,-1)
# print(b)
# print(c)

# step2. 计算乘积
x = a @ a.T
print(x)

# step3. 计算最后的结果
res = x/(b*c)
print(res)

最后可以用torch.cosine_similarity函数检验一下是否计算有误。