Fonction torch.cumulative_trapezoid de PyTorch

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

torch.cumulative_trapezoidC'est une fonction de PyTorch utilisée pour calculer l'intégrale trapézoïdale cumulative d'une fonction. Elle utilise la règle des trapèzes pour calculer l'intégrale cumulative de la fonction aux points donnés.

Définition de fonction

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

Parameters:

  • y(Tensor): Les valeurs de la fonction à intégrer.
  • x(Tensor, optionnel): Les coordonnées x correspondant à la fonction. Si None, la taille du pas dx est utilisée.
  • dx(float, optionnel): La taille du pas entre les coordonnées x. La valeur par défaut est 1.0.
  • dim(int, optionnel): La dimension selon laquelle intégrer. La valeur par défaut est -1.

Valeur de retour:

  • torch.Tensor: Renvoie le résultat de l'intégrale cumulée.

Exemple d'utilisation

Exemple

import torch

# Créer les valeurs de la fonction
y = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])

# Intégrale trapézoïdale cumulée
result = torch.cumulative_trapezoid(y)

print("Valeurs de la fonction y :", y)
print("Résultat de l'intégrale cumulée :", result)

La sortie est :

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

Exemple - Spécification des coordonnées 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("Résultat de l'intégrale cumulée :", result)

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

Autres extensions