import torch.nn.functional as F F.cosin_similarity(a,b,dim=1)
沿着dim维度对a,b两个tensor计算余弦相似度。由于dim属性的存在，使得a,b两个tensor可以为任意维。
import torch.nn.functional as F F.cosin_similarity(a,b,dim=1)
沿着dim维度对a,b两个tensor计算余弦相似度。由于dim属性的存在，使得a,b两个tensor可以为任意维。
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
