小白学Pytorch系列--Torch.nn API Distance Functions(13)

发呆的比目鱼

于 2023-03-29 00:27:28 发布

阅读量159

点赞数

分类专栏： PyTorch框架文章标签： pytorch 人工智能 python

本文链接：https://blog.csdn.net/weixin_42486623/article/details/129773560

版权

52 篇文章

订阅专栏

本文介绍了Pytorch库中的两个函数，nn.CosineSimilarity用于计算向量间的余弦相似度，nn.PairwiseDistance则用于计算向量或矩阵列的成对距离。示例展示了如何使用这两个函数处理随机生成的数据。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

小白学Pytorch系列–Torch.nn API Distance Functions(13)

方法	注释
nn.CosineSimilarity	返回沿着dim计算的x1和x2之间的余弦相似度。
nn.PairwiseDistance	计算输入向量之间或输入矩阵列之间的成对距离。

>>> input1 = torch.randn(100, 128)
>>> input2 = torch.randn(100, 128)
>>> cos = nn.CosineSimilarity(dim=1, eps=1e-6)
>>> output = cos(input1, input2)

计算输入向量之间或输入矩阵列之间的成对距离。

>>> pdist = nn.PairwiseDistance(p=2)
>>> input1 = torch.randn(100, 128)
>>> input2 = torch.randn(100, 128)
>>> output = pdist(input1, input2)