Fonction torch.trapezoid de PyTorch

Pytorch torch 参考手册Manuel de référence PyTorch torch

torch.trapezoidest une fonction de PyTorch utilisée pour calculer des intégrales trapézoïdales. Cette fonction utilise la règle des trapèzes pour effectuer l'intégration numérique d'une fonction donnée.

Définition de fonction

torch.trapezoid(y, x, dx)

Description du paramètre

  • y: les valeurs de la fonction à intégrer
  • x: les valeurs de la variable d'intégration (facultatif)
  • dx: l'espacement d'échantillonnage (utilisé lorsque x n'est pas fourni)

Exemples d'utilisation

Exemple

import torch

# Utiliser dx pour l'intégration trapézoïdale
y = torch.tensor([1.0, 2.0, 3.0, 4.0])

# dx=1.0 indique un espacement d'échantillonnage uniforme
result = torch.trapezoid(y, dx=1.0)

print("Résultat de l'intégrale trapézoïdale :", result)

# Intégrer en utilisant les coordonnées 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("Résultat de l'intégrale en utilisant les coordonnées x :", result2)

Le résultat de sortie est:

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

Pytorch torch 参考手册Manuel de référence PyTorch torch

Autres extensions