【Hackathon 5th No.15】为 Paddle 新增 Tensor.to() 以及 Layer.astype() API (#…

…6305) * add doc_cn tensorto and layerastype * fix ci error * Update docs/api/paddle/Tensor_cn.rst Co-authored-by: zachary sun <70642955+sunzhongkai588@users.noreply.github.com> * Update docs/api/paddle/nn/Layer_cn.rst Co-authored-by: zachary sun <70642955+sunzhongkai588@users.noreply.github.com> * Update docs/api/paddle/nn/Overview_cn.rst Co-authored-by: zachary sun <70642955+sunzhongkai588@users.noreply.github.com> * update doc style * Update docs/api/paddle/nn/Layer_cn.rst Co-authored-by: zachary sun <70642955+sunzhongkai588@users.noreply.github.com> --------- Co-authored-by: zachary sun <70642955+sunzhongkai588@users.noreply.github.com>
PaddlePaddle · Nov 15, 2023 · 30451ba · 30451ba
1 parent 713e0e6
commit 30451ba
Show file tree

Hide file tree

Showing 4 changed files with 39 additions and 1 deletion.
diff --git a/docs/api/paddle/Tensor/Overview_en.rst b/docs/api/paddle/Tensor/Overview_en.rst
@@ -252,6 +252,7 @@ Methods
     tanh
     tanh_
     tile
+    to
     tolist
     topk
     trace

diff --git a/docs/api/paddle/Tensor_cn.rst b/docs/api/paddle/Tensor_cn.rst
@@ -2345,6 +2345,27 @@ tile(repeat_times, name=None)
 
 请参考 :ref:`cn_api_paddle_tile`
 
+to(*args, **kwargs)
+:::::::::
+
+转换 Tensor 的设备或/和数据类型，并且返回转换后的 Tensor。该函数将会从 ``args`` 以及 ``kwargs`` 中解析出要转换到的目标类型 dtype 以及目标设备 place。
+目前支持一下三种方式调用该方法：
+
+    1. to(dtype, blocking=True)
+    2. to(device, dtype=None, blocking=True)
+    3. to(other, blocking=True)
+
+其中， ``dtype`` 可以是 ``paddle.dtype``, ``numpy.dtype`` 类型或者是 ``["bfloat16", "float16", "float32", "float64", "int8", "int16", "int32",
+"int64", "uint8", "complex64", "complex128", "bool"]`` 中的任意一个 ``str``。 ``device`` 可以是 ``paddle.CPUPlace()``, ``paddle.CUDAPlace()``,
+``paddle.CUDAPinnedPlace()``, ``paddle.XPUPlace()``, ``paddle.CustomPlace()`` 或者 ``str``。 ``other`` 需要是 ``Tensor`` 类型。
+
+返回：类型转换后的新的 Tensor
+
+返回类型：Tensor
+
+**代码示例**
+COPY-FROM: paddle.Tensor.to
+
 tolist()
 :::::::::
 

diff --git a/docs/api/paddle/nn/Layer_cn.rst b/docs/api/paddle/nn/Layer_cn.rst
@@ -407,6 +407,22 @@ to(device=None, dtype=None, blocking=None)
 
 COPY-FROM: paddle.nn.Layer.to
 
+astype(dtype=None)
+:::::::::
+
+将 Layer 的所有 ``parameters`` 和 ``buffers`` 的数据类型转换为 ``dtype``，并返回这个 Layer。
+
+**参数**
+
+    - **dtype** (str | paddle.dtype | numpy.dtype) - 转换后的 dtype，str 类型支持"bool", "bfloat16", "float16", "float32", "float64", "int8", "int16", "int32", "int64", "uint8", "complex64", "complex128"。
+
+返回：类型转换后的 Layer
+
+返回类型：Layer
+
+**代码示例**
+COPY-FROM: paddle.nn.Layer.astype
+
 float(excluded_layers=None)
 '''''''''
 

diff --git a/docs/api/paddle/nn/Overview_cn.rst b/docs/api/paddle/nn/Overview_cn.rst
@@ -461,7 +461,7 @@ Embedding 相关函数
     :header: "API 名称", "API 功能"
 
 
-    " :ref:`paddle.nn.functional.diag_embed <cn_api_paddle_nn_functional_diag_embed>` ", "对角线 Embedding 方法"
+    " paddle.nn.functional.diag_embed ", "对角线 Embedding 方法，paddle.nn.functional.diag_embed 已废弃，请使用 :ref:`paddle.diag_embed <cn_api_paddle_diag_embed>` "
     " :ref:`paddle.nn.functional.embedding <cn_api_paddle_nn_functional_embedding>` ", "Embedding 方法"
 
 .. _loss_functional:
-Original file line number
+Diff line change
@@ Expand Up / @@ -252,6 +252,7 @@ Methods @@
         tanh
         tanh_
         tile
+        to
         tolist
         topk
         trace
@@ Expand Down @@