From b29b002bccc272f5b5f5bb14c7a76ddee4f86cc2 Mon Sep 17 00:00:00 2001
From: x54-729 <17307130121@fudan.edu.cn>
Date: Wed, 25 May 2022 17:17:58 +0000
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=20paddle=5Fto=20=E4=B8=AD?=
 =?UTF-8?q?=E5=85=B3=E4=BA=8E=20device=20=E7=B1=BB=E5=9E=8B=E7=9A=84?=
 =?UTF-8?q?=E6=96=87=E6=A1=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 fastNLP/core/utils/paddle_utils.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/fastNLP/core/utils/paddle_utils.py b/fastNLP/core/utils/paddle_utils.py
index f14a2bce..9e7e73a4 100644
--- a/fastNLP/core/utils/paddle_utils.py
+++ b/fastNLP/core/utils/paddle_utils.py
@@ -68,7 +68,8 @@ def paddle_to(data: "paddle.Tensor", device: Union[str, int, 'paddle.fluid.core_
     该函数只是集成了 :func:`paddle.Tensor.cpu` 和 :func:`paddle.Tensor.cuda` 两个函数。
 
     :param data: 要迁移的张量；
-    :param device: 目标设备，可以是 ``str`` 或 ``int`` 类型；
+    :param device: 目标设备，可以是 ``str`` 或 ``int`` 及 **paddle** 自己的 :class:`paddle.fluid.core_avx.Place`、
+        :class:`paddle.CPUPlace` 和 :class:`paddle.CUDAPlace` 类型；
     :return: 迁移后的张量；
     """
     if isinstance(device, paddle.fluid.core_avx.Place):