在Brainstorm中,可以使用自动微分来计算梯度,以下是一个简单的示例代码:
import torch
from torch.autograd import Variable
# 定义变量
x = Variable(torch.tensor(2.0), requires_grad=True)
y = x**2
# 计算梯度
y.backward()
# 打印梯度
print(x.grad)
在这个示例中,我们首先定义了一个变量x,并设置requires_grad=True以便计算梯度。然后我们定义了一个新的变量y,表示x的平方。接着调用y.backward()来计算y对x的梯度,最后打印出x的梯度值。这就是使用自动微分来计算梯度的基本步骤。