Función torch.trapz de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.trapzes la función de PyTorch para calcular la integral trapezoidal, y estorch.trapezoidun alias de la función, ambas tienen la misma funcionalidad.

Definición de la función

torch.trapz(y, x, dx)

Descripción de parámetros

  • y: valores de la función a integrar
  • x: valor de la variable de integración (opcional)
  • dx: espacio de muestreo (se usa cuando no se proporciona x)

Ejemplo de uso

Ejemplo

import torch

# Realizar la integral trapezoidal usando dx
y = torch.tensor([1.0, 2.0, 3.0, 4.0])

# dx=1.0 indica un espacio de muestreo uniforme
result = torch.trapz(y, dx=1.0)

print("Resultado de la integral trapezoidal (trapz):", result)

# Equivalente a trapezoid
result2 = torch.trapezoid(y, dx=1.0)
print("Resultado de la integral trapezoidal (trapezoid):", result2)

El resultado de salida es:

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

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones