蓝灯123

导航

一个稍微复杂的梯度计算例子

多自变量的函数，求梯度 x = torch.randn(3, requires_grad=True) # x 中存了三个变量 x1,x2,x3 y = x + 2 z = y * y * 3 z = z.mean() # 求导 z.backward() print(x.grad) # dz/dx #

posted on 2022-01-20 23:37  蓝灯123  阅读(122)  评论(0)  编辑  收藏  举报