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