基于Dlpack,Cupy.array和pytoch的tensor可相互转化
Cupy.array<-->Dlpack.Tensor<-->torch.Tensor当前安装Cupy需要从源码安装,因为直接pip安装没有接口函数cupy(左边和中间的转换)from cupy.core.dlpack import toDlpackfrom cupy.core.dlpack import fromDlpacktorch(中间和右边的转换...
原创
2018-10-27 09:10:27 ·
1331 阅读 ·
0 评论