从 reshape 到 einops：理解 Transformer 代码中的 Tensor Shape DSLAug 27, 2026·6245 words·13 minsDeep-Learning Einops CS336 PyTorch Transformer Ml-Infra起因 # 最近在做 CS336（Language Modeling from Scratch）的 Transformer 实现，代码里反复出现这种写法：