PairwiseDistance¶
- class torch.nn.PairwiseDistance(p=2.0, eps=1e-06, keepdim=False)[source]¶
计算输入向量之间或输入矩阵的列之间的成对距离。
距离使用
p
-范数计算,并添加常数eps
以避免如果p
为负则出现除以零的情况,即其中 是全1向量,
p
-范数定义如下。- 参数
- 形状
输入1: 或 ,其中 N = 批次维度 且 D = 向量维度
输入2: 或 ,与输入1形状相同
输出:根据输入维度输出 或 。如果
keepdim
为True
,则根据输入维度输出 或 。
- 示例:
>>> pdist = nn.PairwiseDistance(p=2) >>> input1 = torch.randn(100, 128) >>> input2 = torch.randn(100, 128) >>> output = pdist(input1, input2)