下面一段函数,功能是对输入的特征描述子desc1、desc2做余弦相似度的计算。
def calc_cosine_similarity(desc1, desc2):
'''
Input:
desc1: [B,N,*,C]
desc2: [B,N,*,C]
Ret:
similarity: [B,N,*]
'''
inner_product = torch.sum(
torch.mul(desc1, desc2), dim=

最低0.47元/天 解锁文章
1611

被折叠的 条评论
为什么被折叠?



