diff --git a/docs/api/paddle/Tensor/Overview_en.rst b/docs/api/paddle/Tensor/Overview_en.rst index cdee75a9452..11c70b32e0e 100644 --- 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 index a129148f356..a417316187b 100755 --- 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 index 5c34ca467a5..8b89f5789f2 100644 --- 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 index 793e79f60cb..c517e6607b9 100644 --- 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 ` ", "对角线 Embedding 方法" + " paddle.nn.functional.diag_embed ", "对角线 Embedding 方法,paddle.nn.functional.diag_embed 已废弃,请使用 :ref:`paddle.diag_embed ` " " :ref:`paddle.nn.functional.embedding ` ", "Embedding 方法" .. _loss_functional: