tensor-layouts

#tensor-layouts

@fedebruzzone7: 从范畴论的角度：CuTe 布局的范畴基础 https://arxiv.org/abs/2601.05972

X AI KOLs Timeline ↗ · 2026-08-18 缓存

本文引入了一个范畴框架来形式化 NVIDIA 的 CUTLASS 库中的布局代数，定义了范畴和态射以表征张量布局，并提供了一个 Python 实现以及兼容性证明。

0 人收藏 0 人点赞