"""第 10 课补全练习：按权重汇总。修改 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(scores, values):
    raise NotImplementedError("请补全：按权重汇总；参考答案见 solution.py")

y=solve(torch.zeros(3),torch.tensor([[1.,0.],[0.,2.],[2.,2.]]))
torch.testing.assert_close(y,torch.tensor([1.,4/3]))
print("本课补全练习通过。")