o #j @sHddlZddlmZddlmZddlmZddgZddZd dZ dS) N) LoDTensor) check_type)in_dygraph_mode to_dlpack from_dlpackcCsXtr!t|tjtjjjjfstdt|d| St |dt d| 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) >>> # doctest: +SKIP('the address will change in every run') zAThe type of 'x' in to_dlpack must be paddle.Tensor, but received .xr)r isinstancepaddleZTensorbasecoreeager TypeErrortypevalueZ get_tensorZ _to_dlpackrr)r rT/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/utils/dlpack.pyrscCsht|}|jdko |jdk}|stdt|dtr+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=Place(cpu), stop_gradient=True, [[0.20000000, 0.30000001, 0.50000000, 0.89999998], [0.10000000, 0.20000000, 0.60000002, 0.69999999]]) builtinsZ PyCapsulezKThe type of 'dlpack' in from_dlpack must be PyCapsule object, but received r) r __module____name__rrr r r rZ to_tensor)ZdlpacktZ dlpack_flagoutrrrrAs ) r Z base.corerZbase.data_feederrZbase.frameworkr__all__rrrrrrs    &