Функция torch.trapezoid PyTorch

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

torch.trapezoid— это функция в PyTorch, используемая для вычисления интегралов методом трапеций. Эта функция использует метод трапеций для численного интегрирования заданной функции.

Определение функции

torch.trapezoid(y, x, dx)

Описание параметра

y: значения интегрируемой функции
x: значения переменной интегрирования (необязательно)
dx: шаг дискретизации (используется, когда x не указан)

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

Пример

import torch

# Используйте dx для интегрирования методом трапеций
y = torch.tensor([1.0, 2.0, 3.0, 4.0])

# dx=1.0 указывает на равномерный шаг дискретизации
result = torch.trapezoid(y, dx=1.0)

print("Результат интеграла методом трапеций:", result)

# Интегрирование с использованием координат x
x = torch.tensor([0.0, 0.5, 1.0, 1.5])
y2 = torch.tensor([0.0, 0.5, 1.0, 1.5])

result2 = torch.trapezoid(y2, x)
print("Результат интеграла с использованием координат x:", result2)

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

梯形积分结果: tensor(7.5000)
使用 x 坐标的积分结果: tensor(1.2500)