Python PyTorch diff用法及代碼示例

本文簡要介紹python語言中 torch.diff 的用法。

用法:

torch.diff(input, n=1, dim=- 1, prepend=None, append=None) → Tensor

參數：

• input(Tensor) -計算差異的張量

• n(int,可選的) -遞歸計算差異的次數

• dim(int,可選的) -計算差異的維度。默認是最後一個維度。

• prepend(Tensor,可選的) -在計算差異之前，沿 dim 將值添加到 input 或附加到 input。它們的尺寸必須與輸入的尺寸相同，並且它們的形狀必須與輸入的形狀匹配，但 dim 除外。

• append(Tensor,可選的) -在計算差異之前，沿 dim 將值添加到 input 或附加到 input。它們的尺寸必須與輸入的尺寸相同，並且它們的形狀必須與輸入的形狀匹配，但 dim 除外。

關鍵字參數：

out(Tensor,可選的) -輸出張量。

計算沿給定維度的n-th 前向差異。

first-order 差異由 out[i] = input[i + 1] - input[i] 給出。高階差異通過使用torch.diff()遞歸計算。

注意

當前僅支持n = 1

例子：

>>> a = torch.tensor([1, 3, 2])
>>> torch.diff(a)
tensor([ 2, -1])
>>> b = torch.tensor([4, 5])
>>> torch.diff(a, append=b)
tensor([ 2, -1,  2,  1])
>>> c = torch.tensor([[1, 2, 3], [3, 4, 5]])
>>> torch.diff(c, dim=0)
tensor([[2, 2, 2]])
>>> torch.diff(c, dim=1)
tensor([[1, 1],
        [1, 1]])