溫馨提示×

Brainstorm中怎么使用自動(dòng)微分來計(jì)算梯度

小億
82
2024-03-18 13:47:46

在Brainstorm中，可以使用自動(dòng)微分來計(jì)算梯度，以下是一個(gè)簡單的示例代碼：

import torch
from torch.autograd import Variable

# 定義變量
x = Variable(torch.tensor(2.0), requires_grad=True)
y = x**2

# 計(jì)算梯度
y.backward()

# 打印梯度
print(x.grad)

在這個(gè)示例中，我們首先定義了一個(gè)變量x，并設(shè)置requires_grad=True以便計(jì)算梯度。然后我們定義了一個(gè)新的變量y，表示x的平方。接著調(diào)用y.backward()來計(jì)算y對x的梯度，最后打印出x的梯度值。這就是使用自動(dòng)微分來計(jì)算梯度的基本步驟。

0 踩