DLPack开放的内存张量结构

联合创作 · 2023-09-30 16:02

DLPack是一种开放的内存张量结构,用于在框架之间共享张量。 DLPack启用

  • 在深度学习框架之间更轻松地共享操作员。
  • 更容易包装供应商级别的运营商实施,允许在引入新设备/操作时进行协作。
  • 快速交换后端实现,如不同版本的BLAS
  • 对于最终用户来说,这可以带来更多的运营商,并且可以在框架之间混合使用。

我们不打算实现Tensor和Ops,而是将其用作跨框架重用张量和操作的公共桥梁。

浏览 9
点赞
评论
收藏
分享

手机扫一扫分享

编辑 分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

编辑 分享
举报