to_dlpack

paddle.utils.dlpack. to_dlpack ( x ) [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>