最新

嘿，Hacker News，我为自己写了一个用于可视化张量的Python库。这使得学习和调试深度学习代码变得更加容易。它与Colab/Jupyter笔记本以及其他Python环境无缝兼容。这个库是基于图形后端Chalk构建的（[https://chalk-diagrams.github.io/](https://chalk-diagrams.github.io/)）。 <p><i>为什么？</i></p> 理解深度学习代码是困难的——尤其是当它是陌生的，因为很难在脑海中想象张量操作，例如 `F.conv2d(x.unsqueeze(1), w.transpose(-1, -2)).squeeze().view(B, L, -1)`。仅仅打印形状和张量值只能让我了解一部分。tensordiagram让我能够快速绘制张量图。 <p>其他用于创建张量图的Python库要么过于关注物理和数学，不够适合笔记本使用，要么仅限于可视化单个张量，或者服务于更广泛的目的（因此学习曲线陡峭）。</p>

“这是一个开源的[您的工具类型]。我们为个人使用提供慷慨且功能齐全的免费版。我们正在收集关于初始发布的反馈，非常希望听到您对这一方法的看法！”