"""第 04 课补全练习：把目标右移一位。修改 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(ids):
    return ids[:-1], ids[1:]

assert solve([1,2,3,4])==([1,2,3],[2,3,4])
print("本课补全练习通过。")