"""第 20 课补全练习：调节候选分布。修改 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(logits, temperature):
    raise NotImplementedError("请补全：调节候选分布；参考答案见 solution.py")

x=torch.tensor([2.,1.,0.])
a=solve(x,.5);b=solve(x,1.)
assert a[0]>b[0]
assert abs(a.sum().item()-1)<1e-6
print("本课补全练习通过。")