当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python torch.nn.CosineSimilarity用法及代码示例


用法:

class torch.nn.CosineSimilarity(dim=1, eps=1e-08)

参数：

  • dim(int,可选的) -计算余弦相似度的维度。默认值：1

  • eps(float,可选的) -小值以避免被零除。默认值：1e-8

返回 和 之间的余弦相似度，沿 dim 计算。

形状：

  • 输入 1： 其中 D 位于位置 dim

  • 输入 2： ，与 x1 相同的维度数，在维度 dim 匹配 x1 大小，

    并且可以在其他维度上用 x1 广播。

  • 输出：

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

相关用法


注：本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.nn.CosineSimilarity。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。