Функция torch.diff в PyTorch

Pytorch torch 参考手册 Справочное руководство 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: количество разностей, по умолчанию 1
dim: размерность, вдоль которой вычисляются разности
prepend: добавляется в начало input перед вычислением
append: добавляется в конец input после вычисления

Примеры использования

Пример

import torch

# Создать тензор
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)

Результат вывода:

tensor([1, 1])

Пример

import torch

# Создать 2D-тензор
x = torch.tensor([[1, 3, 6, 10], [2, 5, 9, 14]])

# Вычислить разности вдоль последней размерности
y = torch.diff(x)
print(y)

Результат вывода:

tensor([[2, 3, 4],
        [3, 4, 5]])

Pytorch torch 参考手册 Справочное руководство PyTorch torch

Другие расширения