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

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

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

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

torch.trapz(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.trapz(y, dx=1.0)

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

# Эквивалентно trapezoid
result2 = torch.trapezoid(y, dx=1.0)
print("Результат интегрирования методом трапеций (трапеция):", result2)

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

梯形积分结果 (trapz): tensor(7.5000)
梯形积分结果 (trapezoid): tensor(7.5000)

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

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