[ 仅参数名不一致 ]torch.nn.functional.embedding¶
torch.nn.functional.embedding¶
torch.nn.functional.embedding(input, weight, padding_idx=None, max_norm=None, norm_type=2.0, scale_grad_by_freq=False, sparse=False)
paddle.nn.functional.embedding¶
paddle.nn.functional.embedding(x, weight, padding_idx=None, max_norm=None, norm_type=2.0, sparse=False, scale_grad_by_freq=False, name=None)
两者功能一致且参数用法一致,仅参数名不一致,具体如下:
参数映射¶
PyTorch | PaddlePaddle | 备注 |
---|---|---|
input | x | 输入 Tensor,仅参数名不同。 |
weight | weight | 嵌入矩阵权重。 |
padding_idx | padding_idx | 视为填充的下标,参数完全一致。 |
max_norm | max_norm | 如果给定,Embeddding 向量的范数(范数的计算方式由 norm_type 决定)超过了 max_norm 这个界限,就要再进行归一化。 |
norm_type | norm_type | 应用 max_norm 时所计算的 p 阶范数的 p 值。 |
scale_grad_by_freq | scale_grad_by_freq | 是否根据单词在 mini-batch 中出现频率的倒数缩放梯度。 |
sparse | sparse | 是否使用稀疏更新。 |