Brainstorm中怎么使用自动微分来计算梯度

在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的梯度值。这就是使用自动微分来计算梯度的基本步骤。