- 准备工作
import torch
x = torch.arange(4.0)
# 在计算y关于x的梯度前,需要一个地方来储存梯度
x.requires_grad_(True)
也可以直接在创建tensor的时候直接定义requires_grad
torch.arange(4.0, requires_grad=True) <
import torch
x = torch.arange(4.0)
# 在计算y关于x的梯度前,需要一个地方来储存梯度
x.requires_grad_(True)
也可以直接在创建tensor的时候直接定义requires_grad
torch.arange(4.0, requires_grad=True) <