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,支持的设备类型为: CPUCUDAPlaceCUDAPinnedPlace

代码示例 1

COPY-FROM: paddle.utils.dlpack.from_dlpack:code-paddle-from-paddle

代码示例 2

COPY-FROM: paddle.utils.dlpack.from_dlpack:code-paddle-from-numpy