Skip to content

Commit 3c22902

Browse files
cangtianhuangluotao1DrRyanHuang
authored
修复文档 paddle.nn.functional.embedding paddle.index_add (#7250)
Co-authored-by: Tao Luo <luotao02@baidu.com> Co-authored-by: Ryan <zihaohuang@aliyun.com>
1 parent b865711 commit 3c22902

3 files changed

Lines changed: 24 additions & 24 deletions

File tree

docs/api/paddle/index_add_cn.rst

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ index_add
5555
输出:
5656
out.shape = [3, 3]
5757
out.data = [[2., 1., 2.],
58-
[1., 1., 1.],
58+
[2., 1., 2.],
5959
[2., 1., 2.]]
6060

6161
**示例 1 图解说明**
@@ -79,10 +79,10 @@ index_add
7979
参数
8080
:::::::::
8181

82-
- **x** Tensor)– 输入 Tensor。 ``x`` 的数据类型可以是 float16, float32,float64,int32,int64。
83-
- **index** Tensor)– 包含索引下标的 1-D Tensor。数据类型为 int32 或者 int64。
84-
- **axis** (int) 索引轴。数据类型为 int。
85-
- **value** Tensor)– 与 ``x`` 相加的 Tensor。 ``value`` 的数据类型同 ``x`` 。
82+
- **x** (Tensor) - 输入 Tensor。 ``x`` 的数据类型可以是 float16float32,float64,int32,int64。
83+
- **index** (Tensor) - 包含索引下标的 1-D Tensor。数据类型为 int32 或者 int64。
84+
- **axis** (int) - 索引轴。数据类型为 int。
85+
- **value** (Tensor) - 与 ``x`` 相加的 Tensor。 ``value`` 的数据类型同 ``x`` 。
8686
- **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 None。
8787

8888
返回

docs/api/paddle/nn/Embedding_cn.rst

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -5,12 +5,12 @@ Embedding
55

66
.. py:class:: paddle.nn.Embedding(num_embeddings, embedding_dim, padding_idx=None, max_norm=None, norm_type=2.0, *, scale_grad_by_freq=False, sparse=False, _weight=None, _freeze=False, device=None, dtype=None, weight_attr=None, name=None)
77
8-
嵌入层(Embedding Layer),用于构建 ``Embedding`` 的一个可调用对象,具体用法参照 ``代码示例``。其根据 ``x`` 中的 id 信息从 embedding 矩阵中查询对应 embedding 信息,并会根据输入的 size (num_embeddings, embedding_dim)和 weight_attr 自动构造一个二维 embedding 矩阵。
8+
嵌入层(Embedding Layer),用于构建 ``Embedding`` 的一个可调用对象,具体用法参照 ``代码示例`` 。其根据 ``x`` 中的 id 信息从 embedding 矩阵中查询对应 embedding 信息,并会根据输入的 size (num_embeddings, embedding_dim) 和 weight_attr 自动构造一个二维 embedding 矩阵。
99

1010
输出的 Tensor 的 shape 是在输入 Tensor shape 的最后一维后面添加了 embedding_dim 的维度。
1111

1212
.. note::
13-
input 中的 id 必须满足 ``0 <= id < size[0]``,否则程序会抛异常退出
13+
x 中的 id 必须满足 ``0 <= id < num_embeddings`` ,否则程序会抛出异常并退出
1414

1515
.. code-block:: text
1616
@@ -35,17 +35,17 @@ Embedding
3535
参数
3636
::::::::::::
3737

38-
- **num_embeddings** (int) - 嵌入字典的大小,input 中的 id 必须满足 ``0 <= id < num_embeddings`` 。
38+
- **num_embeddings** (int) - 嵌入字典的大小,输入中的 id 必须满足 ``0 <= id < num_embeddings`` 。
3939
- **embedding_dim** (int) - 每个嵌入向量的维度。
40-
- **padding_idx** (int|long|None,可选) - padding_idx 的配置区间为 ``[-weight.shape[0], weight.shape[0])``,如果配置了 padding_idx,那么在训练过程中遇到此 id 时,其参数及对应的梯度将会以 0 进行填充。
40+
- **padding_idx** (int|long|None,可选) - padding_idx 的配置区间为 ``[-num_embeddings, num_embeddings)`` ,如果配置了 padding_idx,那么在训练过程中遇到此 id 时,其参数及对应的梯度将会以 0 进行填充。默认值为 None
4141
- **max_norm** (float,可选) - 若声明,会将范数大于此值的词嵌入向量重新归一化,使其范数等于此值。在动态图模式下会对 ``weight`` 产生 inplace 修改。默认值为 None。
42-
- **norm_type** (float) - 应用 ``max_norm`` 时所计算的 p 阶范数的 p 值。默认值 2.0。
42+
- **norm_type** (float) - 应用 ``max_norm`` 时所计算的 p 阶范数的 p 值。默认值为 2.0。
4343

4444
关键字参数
4545
::::::::::::
4646

47-
- **scale_grad_by_freq** (bool,可选) - 是否根据单词在 mini-batch 中出现频率的倒数缩放梯度。默认值 False。
48-
- **sparse** (bool,可选) - 是否使用稀疏更新,在词嵌入权重较大的情况下,使用稀疏更新能够获得更快的训练速度及更小的内存/显存占用。
47+
- **scale_grad_by_freq** (bool,可选) - 是否根据单词在 mini-batch 中出现频率的倒数缩放梯度。默认值为 False。
48+
- **sparse** (bool,可选) - 是否使用稀疏更新,在词嵌入权重较大的情况下,使用稀疏更新能够获得更快的训练速度及更小的内存/显存占用。默认值为 False。
4949
- **_weight** (Tensor,可选) - 嵌入矩阵的初始化参数,如果提供了此参数,则不会创建新的嵌入矩阵,而是直接使用此参数。
5050
- **_freeze** (bool,可选) - 是否冻结嵌入矩阵的参数,默认值为 False。
5151
- **device** (str|None,可选) - 指定权重所在的设备,默认值为 None。

docs/api/paddle/nn/functional/embedding_cn.rst

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -7,13 +7,13 @@ embedding
77
88
99
10-
嵌入层(Embedding Layer),根据 x 中的 id 信息从 embedding 矩阵中查询对应 embedding 信息,并会根据输入的 size (vocab_size, emb_size)和 dtype 自动构造一个二维 embedding 矩阵
10+
根据 ``x`` 中的 id 信息从 ``weight`` 矩阵中查询对应的嵌入向量
1111

12-
输出的 Tensor 的 shape 是将输入 Tensor shape 后追加一维 emb_size
12+
输出的 Tensor 的 shape 是将输入 Tensor shape 后追加一维 embedding size(即 weight.shape[1])
1313

1414
.. note::
1515

16-
x 中的 id 必须满足 ``0 =< id < size[0]``,否则程序会抛异常退出
16+
x 中的 id 必须满足 ``0 <= id < weight.shape[0]``,否则程序会抛出异常并退出
1717

1818
.. note::
1919
别名支持: 参数名 ``input`` 可替代 ``x``,如 ``input=tensor_x`` 等价于 ``x=tensor_x``。
@@ -26,7 +26,7 @@ embedding
2626
x.data = [[1, 3], [2, 4], [4, 127]]
2727
x.shape = [3, 2]
2828
weight.shape = [128, 16]
29-
输出是 Tensor:
29+
输出是 Tensor
3030
out.shape = [3, 2, 16]
3131
out.data = [[[0.129435295, 0.244512452, ..., 0.436322452],
3232
[0.345421456, 0.524563927, ..., 0.144534654]],
@@ -42,20 +42,20 @@ embedding
4242
::::::::::::
4343

4444

45-
- **x** (Tensor) - 存储 id 信息的 Tensor,数据类型必须为:int32/int64。input 中的 id 必须满足 ``0 =< id < size[0]``
46-
别名: ``input``
47-
- **weight** (Tensor) - 存储词嵌入权重参数的 Tensor,形状为(num_embeddings, embedding_dim)。
48-
- **padding_idx** (int|long|None,可选) - padding_idx 的配置区间为 ``[-weight.shape[0], weight.shape[0]``,如果配置了 padding_idx,那么在训练过程中遇到此 id 时,其参数及对应的梯度将会以 0 进行填充。如果 padding_idx < 0 ,则 padding_idx 将自动转换到 ``weight.shape[0] + padding_idx`` 。如果设置为 "None",则不会对输出产生影响。默认值:None。
45+
- **x** (Tensor) - 存储 id 信息的 Tensor,数据类型必须为:int32/int64。输入中的 id 必须满足 ``0 <= id < weight.shape[0]``。
46+
别名:``input``
47+
- **weight** (Tensor) - 存储词嵌入权重参数的 Tensor,形状为 (num_embeddings, embedding_dim)
48+
- **padding_idx** (int|long|None,可选) - padding_idx 的配置区间为 ``[-weight.shape[0], weight.shape[0])``,如果配置了 padding_idx,那么在训练过程中遇到此 id 时,其参数及对应的梯度将会以 0 进行填充。如果 padding_idx < 0 ,则 padding_idx 将自动转换到 ``weight.shape[0] + padding_idx``。如果设置为 None,则不会对输出产生影响。默认值为 None。
4949
- **max_norm** (float,可选) - 若声明,会将范数大于此值的词嵌入向量重新归一化,使其范数等于此值。在动态图模式下会对 ``weight`` 产生 inplace 修改。默认值为 None。
50-
- **norm_type** (float) - 应用 ``max_norm`` 时所计算的 p 阶范数的 p 值。默认值 2.0。
51-
- **sparse** (bool,可选) - 是否使用稀疏更新,在词嵌入权重较大的情况下,使用稀疏更新(即设置为 True)能够获得更快的训练速度及更小的内存/显存占用。但是一些优化器不支持稀疏更新,例如 :ref:`cn_api_paddle_optimizer_Adadelta` , :ref:`cn_api_paddle_optimizer_Adamax` , :ref:`cn_api_paddle_optimizer_Lamb` 。在这些情况下,稀疏必须为 False。默认值:False。
52-
- **scale_grad_by_freq** (bool,可选) - 是否根据单词在 mini-batch 中出现频率的倒数缩放梯度。默认值 False。
50+
- **norm_type** (float) - 应用 ``max_norm`` 时所计算的 p 阶范数的 p 值。默认值为 2.0。
51+
- **sparse** (bool,可选) - 是否使用稀疏更新,在词嵌入权重较大的情况下,使用稀疏更新(即设置为 True)能够获得更快的训练速度及更小的内存/显存占用。但是一些优化器不支持稀疏更新,例如 :ref:`cn_api_paddle_optimizer_Adadelta` , :ref:`cn_api_paddle_optimizer_Adamax` , :ref:`cn_api_paddle_optimizer_Lamb` 。在这些情况下,稀疏必须为 False。默认值为 False。
52+
- **scale_grad_by_freq** (bool,可选) - 是否根据单词在 mini-batch 中出现频率的倒数缩放梯度。默认值为 False。
5353
- **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 None。
5454

5555

5656
返回
5757
::::::::::::
58-
Tensor, x 映射后得到的 Embedding Tensor,数据类型和权重定义的类型一致。
58+
Tensorx 映射后得到的 Embedding Tensor,数据类型和权重定义的类型一致。
5959

6060

6161
代码示例

0 commit comments

Comments
 (0)