桃子桃子快讯
返回首页
工具

开源神经网络形状校验工具 TenSEY:设计阶段即捕获张量错误

开发者发布 MIT 协议开源工具 TenSEY,提供可视化编辑器,实时校验张量形状、统计参数并估算 FLOPs 与显存,…

2026.07.05 · 周日1 分钟阅读

TenSEY 是一款由独立开发者近期在 GitHub 上开源的神经网络架构可视化校验工具,目标是在模型设计阶段就发现张量维度不匹配等常见错误,避免这些问题在 GPU 训练时才暴露,造成算力浪费。

核心能力

  • 形状校验:自动推断张量在各算子之间的形状变化,捕获残差连接维度不一致、Linear 层输入输出不匹配等典型张量错误。
  • 资源估算:实时统计模型参数量,并估算 FLOPs 与显存(VRAM)占用,方便开发者在动手训练前评估模型规模。
  • 代码导出:可将可视化设计直接导出为可运行的 PyTorch 代码,省去手写样板代码的过程。

支持范围与许可

工具内置 63 种算子,覆盖常见的神经网络层与张量操作。项目采用 MIT 协议开源,代码托管在 GitHub(仓库地址 aarocy/tensey),并提供基于浏览器的可视化编辑器(tensey.vercel.app),无需本地安装即可使用。

适用场景与局限

对于需要快速搭建原型或在论文复现中反复确认张量维度的研究人员与工程师,TenSEY 提供了一种「先验证、后训练」的轻量工作流。但作为个人维护项目,其算子覆盖深度、文档完善度与社区支持相对有限,更适合作为辅助校验手段,而非完全替代人工代码 review。对于生产级复杂模型,仍建议结合 torchinfo 等成熟工具交叉核对。

信源