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


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


用法:

class torch.nn.PairwiseDistance(p=2.0, eps=1e-06, keepdim=False)

参数

  • p(真实的) -规范度。默认值:2

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

  • keepdim(bool,可选的) -确定是否保留向量维度。默认值:假

使用 p-norm 计算向量 之间的成对距离:

形状:
  • 输入 1: 其中 N = batch dimensionD = vector dimension

  • Input2: ,与 Input1 的形状相同

  • 输出: 基于输入维度。

    如果 keepdimTrue ,则 基于输入维度。

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

相关用法


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