PyTorch torch.gradient Funktion
PyTorch torch Referenzhandbuch
torch.gradientist eine Funktion in PyTorch, die zur Berechnung des Gradienten eines Tensors verwendet wird.
Funktionsdefinition
torch.gradient(input, *, dim=None, spacing=1, edge_order=1)
Anwendungsbeispiel
Beispiel
import torch
# Erstelle einen 1D-Tensor
x = torch.tensor([1.0, 2.0, 4.0, 7.0, 11.0])
# Berechne den Gradienten
grad = torch.gradient(x)
print("Gradient:", grad[0])
# Erstelle einen 1D-Tensor
x = torch.tensor([1.0, 2.0, 4.0, 7.0, 11.0])
# Berechne den Gradienten
grad = torch.gradient(x)
print("Gradient:", grad[0])
Das Ausgabeergebnis ist:
梯度: tensor([1., 2., 3., 4.])
Andere Erweiterungen