\;jDddlZddlmZddlmZddlmZddgZdZd Z y) N) LoDTensor) check_type)in_dygraph_mode to_dlpack from_dlpackcrtrt|tjtjj j jfstdt|d|jjjSt|dtd|jS)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) >>> # doctest: +SKIP('the address will change in every run') zAThe type of 'x' in to_dlpack must be paddle.Tensor, but received .xr)r isinstancepaddleTensorbasecoreeager TypeErrortypevalue get_tensor _to_dlpackrr)r s \G:\00. PROJECTS\API\Inventory\templateJSON\kerjaOCR\Lib\site-packages\paddle/utils/dlpack.pyrrs2!fmmV[[-=-=-C-C-J-JKL!!%a ,  wwy##%0022q# K0 <<>cvt|}|jdk(xr|jdk(}|stdt|dt r@t j jj|}t j|}|St j jj|}|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=Place(cpu), stop_gradient=True, [[0.20000000, 0.30000001, 0.50000000, 0.89999998], [0.10000000, 0.20000000, 0.60000002, 0.69999999]]) builtins PyCapsulezKThe type of 'dlpack' in from_dlpack must be PyCapsule object, but received r ) r __module____name__rrr rrr to_tensor)dlpackt dlpack_flagouts rrrAs6 V A,,*,J{1JK  !&\N! -  kk**62s# ++   & &v .C Jr) r base.corerbase.data_feederrbase.frameworkr__all__rrrrr(s-!),  #L)r