Fonction torch.gradient de PyTorch
Manuel de référence Pytorch torch
torch.gradientest une fonction de PyTorch utilisée pour calculer le gradient d'un tenseur.
Définition de fonction
torch.gradient(input, *, dim=None, spacing=1, edge_order=1)
Exemple d'utilisation
Exemple
import torch
# Créer un tenseur 1D
x = torch.tensor([1.0, 2.0, 4.0, 7.0, 11.0])
# Calculer le gradient
grad = torch.gradient(x)
print("Gradient:", grad[0])
# Créer un tenseur 1D
x = torch.tensor([1.0, 2.0, 4.0, 7.0, 11.0])
# Calculer le gradient
grad = torch.gradient(x)
print("Gradient:", grad[0])
Le résultat de sortie est:
梯度: tensor([1., 2., 3., 4.])
Autres extensions