diff --git a/python/paddle/utils/dlpack.py b/python/paddle/utils/dlpack.py index d9821698f08612d5a69b21a518151e94f63d5694..c818d16125eedd204698ec51b312381c619cb695 100644 --- a/python/paddle/utils/dlpack.py +++ b/python/paddle/utils/dlpack.py @@ -39,13 +39,14 @@ def to_dlpack(x): 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) - # + >>> 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') + """ if in_dygraph_mode(): @@ -76,16 +77,16 @@ def from_dlpack(dlpack): 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]]) + >>> 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]]) """ t = type(dlpack)