一个稍微复杂的梯度计算例子
摘要:
多自变量的函数,求梯度 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 @ 2022-01-20 23:37 蓝灯123 阅读(127) 评论(0) 推荐(0) 编辑