Fonction torch.trapz de PyTorch

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

torch.trapzest une fonction dans PyTorch utilisée pour calculer l'intégration trapézoïdale, et esttorch.trapezoidun alias de la fonction, les deux ont la même fonctionnalité.

Définition de fonction

torch.trapz(y, x, dx)

Description du paramètre

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

Exemples d'utilisation

Exemple

import torch

# Effectuer une intégration trapézoïdale à l'aide de dx
y = torch.tensor([1.0, 2.0, 3.0, 4.0])

# dx=1.0 signifie un pas d'échantillonnage uniforme
result = torch.trapz(y, dx=1.0)

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

# Équivalent à trapezoid
result2 = torch.trapezoid(y, dx=1.0)
print("Résultat de l'intégration trapézoïdale (trapèze) :", result2)

Le résultat de sortie est:

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

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

Autres extensions