婴儿版训练GPTimport numpy as np# ============================== # 1. 固定词典(你建立的字典) # ============================== vocab = {"我":0, "喜":1, "欢":2, "中":3, "国":4, "美":5, "食":6, "[END]":7} idx2word = {v:k for k,v in vocab.items()} vocab_size = len(vocab) d_model = 8 # 向量维度