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.topk_accuracy