













在流程图中清晰标注张量形状和维度语义的关键是建立结构化标注规则,既要体现数值变换又要保留语义信息。以下是具体建议:
建议格式:[维度1:含义1 | 维度2:含义2 (...)]
示例:[B:32 | C:3 | H:224 | W:224]
维度增减操作 (unsqueeze/squeeze)
# Before: [B:32, C:3, H:224]
unsqueeze(-1) → [B:32, C:3, H:224, W:1 (new)]
维度重组操作 (view/reshape)
# Before: [B:32, L:512, D:768]
view(32, 16, 32, 768) → [B:32, H:16, S:32, D:768 (merged L=H×S)]
维度交换操作 (transpose/permute)
# Before: [B:32, C:3, H:224, W:224]
transpose(1,3) → [B:32, W:224, H:224, C:3 (swapped)]
复合维度表示:使用⊗符号表示组合维度
[B:32, T⊗S:10×64 (T=time_steps, S=seq_len)]
动态维度标注:用*表示动态维度
[B:*, C:3, H:?, W:? (dynamic resolution)]
颜色编码方案:
[Input] → [Conv2d] → [Reshape] → [Transpose] → [Output]
│ │ │ │
▼ ▼ ▼ ▼
[B:32 [B:32 [B:32 [B:32
C:3 C:64 H⊗W:224×224 D:512
H:224 H:224 → D:3136] H:224
W:224] W:224] (swapped W↔C)]
Dimension Glossary:
B - Batch Size H - Height
W - Width C - Channels
D - Embedding Dim T - Time Steps
这种标注方案既能清晰展示数值维度的变化轨迹,又能保留维度的语义信息,特别适合需要同时关注数学变换和业务含义的场景(如多模态数据处理)。实际使用时可根据复杂度选择精简版或完整版标注。
此内容由惯性聚合(RSS阅读器)自动聚合整理,仅供阅读参考。 原文来自 — 版权归原作者所有。