megengine.functional.nn.embedding

embedding(inp, weight, padding_idx=None, max_norm=None, norm_type=None)[源代码]

应用查找表进行 embedding。

参数
  • inp (Tensor) – 带有索引的张量。

  • weight (Tensor) – 嵌入的可学习的词向量矩阵。

  • padding_idx (Optional[int]) – 应设置为None,当前尚不支持。

  • max_norm (Optional[float]) – 应设置为None,当前尚不支持。

  • norm_type (Optional[float]) – 应设置为None,当前尚不支持。

返回

输出张量。

更多信息参见 Embedding