dmlc/dlpack

解决的问题

DLPack 提供了一种标准化的开源内存张量结构。这解决了张量被锁定在特定深度学习框架中的问题，使得在不同库之间共享操作符或交换后端实现变得困难。

工作原理

DLPack 不实现自己的张量或操作符，而是作为通用桥梁。它定义了在内存中表示张量的标准化方式，使不同框架能够交换数据，而无需复杂的转换或冗余复制。

适用人群

专为深度学习框架开发者以及需要在不同软件生态系统间共享张量和操作符的厂商级操作符实现开发者设计。

主要亮点

• 用于跨框架共享的标准化内存张量结构。
• 简化厂商级操作符实现的封装。
• 支持快速切换后端实现，例如不同的 BLAS 版本。
• 允许最终用户混合搭配来自不同框架的操作符。