"""第 11 课补全练习：计算三组表示。修改 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(x, wq, wk, wv):
    raise NotImplementedError("请补全：计算三组表示；参考答案见 solution.py")

x=torch.eye(2)
a,b,c=solve(x,x,2*x,3*x)
assert torch.equal(a,x) and torch.equal(b,2*x) and torch.equal(c,3*x)
print("本课补全练习通过。")