torch.diff

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

Вычисляет n-ую прямую разность по заданному измерению.

Разности первого порядка задаются out[i] = input[i + 1] - input[i]. Разности высших порядков вычисляются, используя torch.diff() рекурсивно.

Параметры

• input (Tensor) – тензор, для которого вычисляются разности
• n (int, optional) – количество рекурсивных вычислений разности
• dim (int, optional) – измерение, по которому вычисляется разность. По умолчанию — последнее измерение.
• prepend (Tensor, optional) – значения для добавления в начало или конец input по dim перед вычислением разности. Их размерности должны быть эквивалентны размерности input, а их формы должны соответствовать форме input, за исключением dim.
• append (Tensor, optional) – значения для добавления в начало или конец input по dim перед вычислением разности. Их размерности должны быть эквивалентны размерности input, а их формы должны соответствовать форме input, за исключением dim.

Аргументы по ключевым словам

out (Tensor, optional) – выходной тензор.

Пример:

>>> 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]])