from mxnet import autograd,nd
x=nd.arange(4).reshape((4,1))
print(x)
x.attach_grad()
with autograd.record():
y=2*nd.dot(x.T,x)
y.backward()
print(x.grad)
mxnet自动求梯度
最新推荐文章于 2022-09-30 13:13:09 发布
from mxnet import autograd,nd
x=nd.arange(4).reshape((4,1))
print(x)
x.attach_grad()
with autograd.record():
y=2*nd.dot(x.T,x)
y.backward()
print(x.grad)