diff --git a/python/paddle/incubate/tensor/manipulation.py b/python/paddle/incubate/tensor/manipulation.py
index 4e1380a0bd7237..70c017a8eec363 100644
--- a/python/paddle/incubate/tensor/manipulation.py
+++ b/python/paddle/incubate/tensor/manipulation.py
@@ -40,12 +40,14 @@ def _npu_identity(x, format=-1):
     Examples:
         .. code-block:: python
 
-            # required: npu
-            import paddle
+            >>> # doctest: +REQUIRES(env:NPU)
+            >>> import paddle
+            >>> paddle.device.set_device('npu')
 
-            x = paddle.ones(shape=[6])
-            y = paddle.incubate._npu_identity(x, 3) # ACL_FORMAT_NC1HWC0 = 3
-            # y.shape = [1, 1, 1, 1, 16]
+            >>> x = paddle.ones(shape=[6])
+            >>> y = paddle.incubate._npu_identity(x, 3) # ACL_FORMAT_NC1HWC0 = 3
+            >>> print(y.shape)
+            [1, 1, 1, 1, 16]
     """
     if in_dynamic_mode():
         return _C_ops.npu_identity(x, format)