"""第 07 课补全练习：根据梯度更新一次。修改 solve，保持下方检查不变。"""
import math, io, argparse
import torch
from torch import nn
from torch.nn import functional as F
torch.set_num_threads(2)
torch.manual_seed(26)

def solve(weight, gradient, rate):
    return weight - rate * gradient

assert abs(solve(0.,-6.,.2)-1.2)<1e-9
print("本课补全练习通过。")