余弦相似度¶
- class torch.nn.CosineSimilarity(dim=1, eps=1e-08)[source]¶
返回 和 之间的余弦相似度,沿 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)