Función PyTorch torch.cumulative_trapezoid

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

torch.cumulative_trapezoidEs una función de PyTorch que se utiliza para calcular la integral trapezoidal acumulativa de una función. Usa la regla del trapecio para calcular la integral acumulativa de la función en puntos dados.

Definición de la función

torch.cumulative_trapezoid(y, x=None, dx=1.0, dim=-1)

Parámetros:

  • y(Tensor): Los valores de la función a integrar.
  • x(Tensor, opcional): Las coordenadas x correspondientes a la función. Si es None, se utiliza el paso dx.
  • dx(float, opcional): El paso entre las coordenadas x. Por defecto es 1.0.
  • dim(int, opcional): La dimensión de la integración. Por defecto es -1.

Valor de retorno:

  • torch.Tensor: Devuelve el resultado de la integral acumulativa.

Ejemplo de uso

Ejemplo

import torch

# Crear los valores de la función
y = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])

# Integral trapezoidal acumulativa
result = torch.cumulative_trapezoid(y)

print("Valores de la función y:", y)
print("Resultado de la integral acumulativa:", result)

El resultado de salida es:

函数值 y: tensor([1., 2., 3., 4., 5.])
累积积分结果: tensor([0.0000, 1.5000, 3.5000, 6.0000, 9.0000])

Ejemplo - Especificar la coordenada 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("Resultado de la integral acumulativa:", result)

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

Otras extensiones