功能描述

算子功能

将整数索引映射到一个固定大小的向量。

算子公式

其中：

• x是输入的整数索引。
• W是一个二维张量，其形状为[num_embeddings, embedding_dim]，其中每一行表示一个嵌入向量。
• W[x]表示从权重矩阵W中取出索引为x的行作为嵌入向量。