[torch 参数更多]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, sparse=False, name=None)
PyTorch 相比 Paddle 支持更多其他参数,具体如下:
参数映射¶
PyTorch | PaddlePaddle | 备注 |
---|---|---|
input | x | 输入 Tensor,仅参数名不同。 |
weight | weight | 嵌入矩阵权重。 |
padding_idx | padding_idx | 视为填充的下标,参数完全一致。 |
max_norm | - | 重新归一化的最大范数,参数不一致,暂无转写方式。 |
norm_type | - | Paddle 无此参数,参数不一致,暂无转写方式。 |
scale_grad_by_freq | - | 按词频进行梯度缩放的比例,参数不一致,暂无转写方式。 |
sparse | sparse | 是否使用稀疏更新。 |