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，当前尚不支持。

Refer to Embedding for more information.

上一页

megengine.functional.nn.indexing_one_hot

下一页

megengine.functional.nn.l1_loss