Pytorch torch.norm, torch.cosine_similarity 对向量或者张量计算Cosine类似度, 欧式距离

torch.cosine_similarity 能够对两个向量或者张量计算类似度html >>> input1 = torch.randn(100, 128) >>> input2 = torch.randn(100, 128) >>> output = torch.cosine_similarity(input1, input2, dim=1) print(output.shape)函数 tor

>>阅读原文<<