Функция torch.gradient в PyTorch
Справочное руководство Pytorch torch
torch.gradientявляется функцией в PyTorch, используемой для вычисления градиента тензора.
Определение функции
torch.gradient(input, *, dim=None, spacing=1, edge_order=1)
Пример использования
Пример
import torch
# Создать одномерный тензор
x = torch.tensor([1.0, 2.0, 4.0, 7.0, 11.0])
# Вычислить градиент
grad = torch.gradient(x)
print("Градиент:", grad[0])
# Создать одномерный тензор
x = torch.tensor([1.0, 2.0, 4.0, 7.0, 11.0])
# Вычислить градиент
grad = torch.gradient(x)
print("Градиент:", grad[0])
Результат вывода:
梯度: tensor([1., 2., 3., 4.])
Другие расширения