smalldiffusion 模型：model_dit.py
本文件实现了 DiT (Peebles & Xie, 2022) 架构，一种基于 Transformer 的扩散模型。
5.1 模块结构
PyTorch Tensor 工具函数技术文档：创建、计算、拼接与索引
总览 PyTorch 张量运算知识结构；创建/reshape/cat/stack/split；逐元与矩阵运算；归约；索引、gather/scatter；比较与逻辑；einsum 等工具；速查与延伸。
Flow Matching Guide and Code 第5章解读：Non-Euclidean Flow Matching
第5章 Non-Euclidean Flow Matching 解读：从动机与黎曼流形设定出发，说明流形上的流、概率路径与连续性方程，边际化技巧（定理10）、RCFM 损失（定理11），以及测地线条件流与基于预度量的条件流；并对照欧氏 FM 与代码8（球面测地线 RCFM）做小结。