Topic
#pytorch
按主题聚合的新闻视图。
Topic Feed
主题：pytorch
共 2 条
PyTorch 中的注意力性能分析
Hugging Face 的《Profiling in PyTorch》系列继续推出第 3 部分，文章演示了如何分析一个朴素的因果注意力实现，并将其与更优化的注意力路径进行对比。文章展示了不同 PyTorch 写法下，注意力相关操作在 profiler trace 中的表现，以及可能多出来的内核。
PyTorch 性能分析入门指南
Hugging Face 发布了“Profiling in PyTorch”系列的第一篇，重点介绍 torch.profiler。文章通过一个简单的矩阵乘法加偏置示例，带新手学习如何阅读性能分析轨迹。