PyTorch torch.diff 函数
Pytorch torch 参考手册torch.diff 是 PyTorch 中用于计算张量相邻元素之间差值的函数。计算公式为 out[i] = input[i+1] - input[i]。
函数定义
torch.diff(input, n=1, dim=-1, prepend=None, append=None)
参数说明：
input: 输入张量n: 差分次数，默认为1dim: 计算差分的维度prepend: 在计算前 prepend 到输入append: 在计算后 append 到输入
使用示例
实例
import torch
# 创建张量
x = torch.tensor([1, 3, 6, 10])
# 计算一阶差分
y = torch.diff(x)
print(y)
# 创建张量
x = torch.tensor([1, 3, 6, 10])
# 计算一阶差分
y = torch.diff(x)
print(y)
输出结果为：
tensor([2, 3, 4])
实例
import torch
# 创建张量
x = torch.tensor([1, 3, 6, 10])
# 计算二阶差分
y = torch.diff(x, n=2)
print(y)
# 创建张量
x = torch.tensor([1, 3, 6, 10])
# 计算二阶差分
y = torch.diff(x, n=2)
print(y)
输出结果为：
tensor([1, 1])
实例
import torch
# 创建二维张量
x = torch.tensor([[1, 3, 6, 10], [2, 5, 9, 14]])
# 对最后一个维度计算差分
y = torch.diff(x)
print(y)
# 创建二维张量
x = torch.tensor([[1, 3, 6, 10], [2, 5, 9, 14]])
# 对最后一个维度计算差分
y = torch.diff(x)
print(y)
输出结果为：
tensor([[2, 3, 4],
[3, 4, 5]])