當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。