从 reshape 到 einops：理解 Transformer 代码中的 Tensor Shape DSL2026-08-27·13 分钟Deep-Learning Einops CS336 Pytorch Transformer起因 # 最近在做 CS336（Language Modeling from Scratch）的 Transformer 实现，代码里反复出现这种写法：
