import torch.nn.functional as F
F.cosine_similarity(a,b,dim=1)
沿着dim
维度对a,b两个tensor计算余弦相似度。由于dim
属性的存在,使得a,b两个tensor可以为任意维。
import torch.nn.functional as F
F.cosine_similarity(a,b,dim=1)
沿着dim
维度对a,b两个tensor计算余弦相似度。由于dim
属性的存在,使得a,b两个tensor可以为任意维。