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 获取更多信息。