研究人员推出了 DanLing NestedTensor,这是一种用于 PyTorch 的新型张量抽象,旨在更有效地处理深度学习中可变大小的输入。这种抽象使多不规则结构成为张量本身的固有属性,减少了填充造成的计算浪费并提高了性能。基准测试显示,与传统的填充方法相比,对于包括 BERT 和 FCN 主干在内的各种深度学习模型,速度显著加快,并且在高变化批次中内存分配明显减少。 AI
影响 这种新的张量抽象可能导致处理可变长度序列的模型训练和推理效率更高,从而有可能降低硬件成本并加速开发。
排序理由 该集群包含一篇详细介绍深度学习计算新技术方法的论文。[lever_c_demoted from research: ic=1 ai=1.0]
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →