"""第 05 课补全练习：变换每个位置的表示。修改 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, weights):
    raise NotImplementedError("请补全：变换每个位置的表示；参考答案见 solution.py")

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