Функция torch.cumulative_trapezoid в PyTorch
Справочное руководство PyTorch torch
torch.cumulative_trapezoidЭто функция в PyTorch, используемая для вычисления кумулятивного интеграла функции методом трапеций. Она использует метод трапеций для вычисления кумулятивного интеграла функции в заданных точках.
Определение функции
torch.cumulative_trapezoid(y, x=None, dx=1.0, dim=-1)
Parameters:
y(Tensor): Значения функции для интегрирования.x(Tensor, optional): Координаты x, соответствующие функции. Если None, используется размер шага dx.dx(float, optional): Размер шага между координатами x. По умолчанию 1.0.dim(int, optional): Размерность, по которой выполняется интегрирование. По умолчанию -1.
Возвращаемое значение:
torch.Tensor: Возвращает результат кумулятивного интеграла.
Пример использования
Пример
import torch
# Создать значения функции
y = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])
# Кумулятивный интеграл методом трапеций
result = torch.cumulative_trapezoid(y)
print("Значения функции y:", y)
print("Результат кумулятивного интеграла:", result)
# Создать значения функции
y = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])
# Кумулятивный интеграл методом трапеций
result = torch.cumulative_trapezoid(y)
print("Значения функции y:", y)
print("Результат кумулятивного интеграла:", result)
Вывод:
函数值 y: tensor([1., 2., 3., 4., 5.]) 累积积分结果: tensor([0.0000, 1.5000, 3.5000, 6.0000, 9.0000])
Пример - Указание координат x
import torch
y = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])
x = torch.tensor([0.0, 0.5, 1.0, 1.5, 2.0])
result = torch.cumulative_trapezoid(y, x)
print("Результат кумулятивного интеграла:", result)
y = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])
x = torch.tensor([0.0, 0.5, 1.0, 1.5, 2.0])
result = torch.cumulative_trapezoid(y, x)
print("Результат кумулятивного интеграла:", result)
Другие расширения