pytorch 计算cosine余弦相似度

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

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