File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
docs/api/paddle/static/nn Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -16,7 +16,7 @@ embedding
1616输出的 Tensor 的 shape 是将输入 Tensor shape 的会在输出的 embedding 最后追加一维 emb_size。
1717
1818.. note ::
19- input 中的 id 必须满足 ``0 =< id < size[0] ``,否则程序会抛异常退出。
19+ input 中的 id 必须满足 ``0 =< id < size[0] `` ,否则程序会抛异常退出。
2020
2121
2222::
@@ -66,7 +66,7 @@ embedding
6666 - **size ** (tuple|list) - embedding 矩阵的维度。必须包含两个元素,第一个元素为 vocab_size(词表大小),第二个为 emb_size(embedding 层维度)。
6767 - **is_sparse ** (bool,可选) - 是否使用稀疏的更新方式,这个参数只会影响反向的梯度更新的性能,sparse 更新速度更快,推荐使用稀疏更新的方式。但某些 optimizer 不支持 sparse 更新,比如 :ref: `cn_api_paddle_optimizer_Adadelta ` 、 :ref: `cn_api_paddle_optimizer_Adamax `,此时 is_sparse 必须为 False。默认为 False。
6868 - **is_distributed ** (bool,可选) - 是否使用分布式的方式存储 embedding 矩阵,仅在多机分布式 cpu 训练中使用。默认为 False。
69- - **padding_idx ** (int|long|None,可选) - padding_idx 需在区间 ``[-vocab_size, vocab_size) ``,否则不生效,``padding_idx < 0 `` 时,padding_idx 会被改成``vocab_size + padding_idx``,input 中等于 padding_index 的 id 对应的 embedding 信息会被设置为 0,且这部分填充数据在训练时将不会被更新。如果为 None,不作处理,默认为 None。
69+ - **padding_idx ** (int|long|None,可选) - padding_idx 需在区间 ``[-vocab_size, vocab_size) `` ,否则不生效, ``padding_idx < 0 `` 时,padding_idx 会被改成 ``vocab_size + padding_idx `` ,input 中等于 padding_index 的 id 对应的 embedding 信息会被设置为 0,且这部分填充数据在训练时将不会被更新。如果为 None,不作处理,默认为 None。
7070 - **param_attr ** (ParamAttr,可选) - 指定权重参数属性的对象。默认值为 None,表示使用默认的权重参数属性。具体用法请参见 :ref: `cn_api_paddle_ParamAttr `。此外,可以通过 ``param_attr `` 参数加载用户自定义或预训练的词向量。只需将本地词向量转为 numpy 数据格式,且保证本地词向量的 shape 和 embedding 的 ``size `` 参数一致,然后使用 :ref: `cn_api_paddle_to_tensor ` 进行初始化,即可实现加载自定义或预训练的词向量。
7171 - **dtype ** (str|paddle.dtype|np.dtype,可选) - 输出 Tensor 的数据类型,数据类型必须为:float32 或 float64,默认为 float32。
7272
You can’t perform that action at this time.
0 commit comments