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

Python PyTorch PairwiseDistance用法及代码示例

本文简要介绍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 dimension 和 D = vector dimension

  • Input2： 或 ，与 Input1 的形状相同

  • 输出： 或 基于输入维度。

    如果 keepdim 是 True ，则 或 基于输入维度。

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

相关用法


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