o Qe® ã@sPddlZddlmZddlmZddlmZmZmZddgZ dd„Z d d„Z dS) éNé)Ú LoDTensor)Ú_non_static_mode)Ú check_typeÚ check_dtypeÚ convert_dtypeÚ to_dlpackÚ from_dlpackcCsVtƒr t|tjtjjjjfƒstd t |ƒ¡ƒ‚|  ¡  ¡  ¡St |dtdƒ|  ¡S)a¿ Encodes a tensor to DLPack. Args: 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: .. code-block:: python 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) # zDThe type of 'x' in to_dlpack must be paddle.Tensor, but received {}.Úxr)rÚ isinstanceÚpaddleZTensorÚfluidÚcoreÚeagerÚ TypeErrorÚformatÚtypeÚvalueZ get_tensorZ _to_dlpackrr)r ©rúCD:\Projects\ConvertPro\env\Lib\site-packages\paddle/utils/dlpack.pyrs þcCsft|ƒ}|jdko |jdk}|std t|ƒ¡ƒ‚tƒr*tjj  |¡}t  |¡}|Stjj  |¡}|S)a§ Decodes a DLPack to a tensor. Args: dlpack (PyCapsule): a PyCapsule object with the dltensor. Returns: out (Tensor): a tensor decoded from DLPack. One thing to be noted, if we get an input dltensor with data type as `bool`, we return the decoded tensor as `uint8`. Examples: .. code-block:: python 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) x = paddle.utils.dlpack.from_dlpack(dlpack) print(x) # Tensor(shape=[2, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, # [[0.20000000, 0.30000001, 0.50000000, 0.89999998], # [0.10000000, 0.20000000, 0.60000002, 0.69999999]]) ÚbuiltinsZ PyCapsulezNThe type of 'dlpack' in from_dlpack must be PyCapsule object, but received {}.) rÚ __module__Ú__name__rrrr r rr Z to_tensor)ZdlpackÚtZ dlpack_flagÚoutrrrr >s þ ) r Z fluid.corerZfluid.frameworkrZfluid.data_feederrrrÚ__all__rr rrrrÚs  þ $