from_dlpack¶
- paddle.utils.dlpack. from_dlpack ( dlpack ) ¶
将 DLPack 格式的 Tensor 解码为 Paddle Tensor,DLPACK 是一种通用稳定的内存数据结构,可用于不同深度学习框架之间的 Tensor 共享。
注解
一个 dlpack 只能被 from_dlpack
解码一次。
参数¶
dlpack (SupportDLPack | PyCapsule) - 一个实现了
__dlpack__
与__dlpack_device__
方法的对象,或者是一个带有 dltensor 的PyCapsule
对象。
返回¶
out (Tensor) - 从 dlpack 中解码得到的 Paddle Tensor,支持的数据类型为: bool,float16,float32,float64,uint8,int8,int16,int32,int64,complex64,complex128,支持的设备类型为:
CPU
,CUDAPlace
,CUDAPinnedPlace
。
代码示例 1¶
COPY-FROM: paddle.utils.dlpack.from_dlpack:code-paddle-from-paddle
代码示例 2¶
COPY-FROM: paddle.utils.dlpack.from_dlpack:code-paddle-from-numpy