蓝灯123

导航

梯度计算简单例子

下面a56爆大奖在线娱乐们来看一个简单的例子, 定义x, y, z三个变量, 其中x是需要计算梯度的. x = torch.tensor([1.0], requires_grad=True) y = torch.tensor([2.0]) z = torch.tensor([2.0]) # 定义运算 f1 = 2*x

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