"""第 06 课补全练习：给正确字评分。修改 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, targets):
    return F.cross_entropy(logits, targets)

assert abs(solve(torch.tensor([[2.,1.,0.]]),torch.tensor([0])).item()-0.40760595)<1e-6
print("本课补全练习通过。")