Función PyTorch torch.gradient
Manual de referencia de Pytorch torch
torch.gradientEs una función en PyTorch utilizada para calcular el gradiente de tensores.
Definición de la función
torch.gradient(input, *, dim=None, spacing=1, edge_order=1)
Ejemplo de uso
Ejemplo
import torch
# Crear un tensor unidimensional
x = torch.tensor([1.0, 2.0, 4.0, 7.0, 11.0])
# Calcular el gradiente
grad = torch.gradient(x)
print("Gradiente:", grad[0])
# Crear un tensor unidimensional
x = torch.tensor([1.0, 2.0, 4.0, 7.0, 11.0])
# Calcular el gradiente
grad = torch.gradient(x)
print("Gradiente:", grad[0])
El resultado de salida es:
梯度: tensor([1., 2., 3., 4.])
Otras extensiones