自動微分(関数の計算結果)
backwardメソッドは、関数を用いた計算の結果にも適用することができる。
code:autograd.py
import torch as pt
def func(x):
return x**2 - 2*x - 3
x0 = 2.0
x = pt.tensor(x0, dtype=pt.float, requires_grad=True)
y = func(x) # 上で定義した関数
y.backward()
print('x:', x)
print('xの勾配:\n', x.grad)
/icons/hr.icon
※ ブラウザのバックボタンで戻る