to_dlpack

paddle.utils.dlpack. to_dlpack ( x ) [源代码]

将 Tensor 对象转化为 DLPack。其中,DLPack 是一种开放的内存张量结构,可用于不同深度学习框架之间的张量共享。

参数

  • x (Tensor) - Paddle Tensor,并且其数据类型为支持 bool,float16,float32,float64,int8,int16,int32,int64,uint8,complex64,complex128。

返回

  • dlpack (PyCapsule) - DLPack,即带有 dltensor 的 PyCapsule 对象。

代码示例

import paddle
# x is a tensor with shape [2, 4]
x = paddle.to_tensor([[0.2, 0.3, 0.5, 0.9],
                      [0.1, 0.2, 0.6, 0.7]])
dlpack = paddle.utils.dlpack.to_dlpack(x)
print(dlpack)
# <capsule object "dltensor" at 0x7f6103c681b0>