to_dlpack¶

paddle. to_dlpack ( x: Tensor ) → CapsuleType [source]

Encodes a tensor to DLPack.

Parameters

x (Tensor) – The input tensor, and the data type can be bool, float16, float32, float64, int8, int16, int32, int64, uint8, complex64, complex128.

Returns

dltensor, and the data type is PyCapsule.

Examples

>>> 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>
>>> 

>>> # dlpack capsule will be renamed to 'used_dltensor' after decoded
>>> y = paddle.utils.dlpack.from_dlpack(dlpack)
>>> print(dlpack)
>>> 
<capsule object "used_dltensor" at 0x7f6103c681b0>

>>> 
>>> # type: ignore
>>> # convert tensor from paddle to other framework using to_dlpack
>>> import torch

>>> x = paddle.randn([2, 4]).to(device="cpu")
>>> y = torch.from_dlpack(paddle.utils.dlpack.to_dlpack(x))
>>> print(y.shape)
torch.Size([2, 4])
>>>