backwardによる勾配ベクトルの計算【torch】
tensor.backward()
tensor ... requires_grad=True を設定した pt.float型、int 型ではダメ。
code:p.py
import torch as pt
def func(x):
return pt.det(x)
x = pt.tensor(1, 0],[2, 3, dtype=pt.float, requires_grad=True)
y = func(x)
y.backward()
print('# by backward:\n', x.grad)
'''
# by jacobian:
# by backward:
'''