Embedding

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	×
Atlas A2 训练系列产品/Atlas A2 推理系列产品	×
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	×
Atlas 训练系列产品	×

功能说明

• 算子功能：把数据集合映射到向量空间，进而将数据进行量化。embedding的二维权重张量为weight(m+1行，n列)，对于任意输入索引张量indices（如1行3列），输出out是一个3行n列的张量，如下所示：

$$\begin{array}{rl}& weight=\left[\begin{array}{cccc}{x}_{01}& x_{02}& ...& x_{0n}\\ x_{11}& x_{12}& ...& x_{1n}\\ x_{21}& x_{22}& ...& x_{2n}\\ ...& & & \\ x_{m1}& x_{m2}& ...& x_{mn}\end{array}\right]\\ & indices=\left[\begin{array}{c}0,2,m\end{array}\right]\\ & out=\left[\begin{array}{cccc}{x}_{01}& x_{02}& ...& x_{0n}\\ x_{21}& x_{22}& ...& x_{2n}\\ x_{m1}& x_{m2}& ...& x_{mn}\end{array}\right]\end{array}$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
weight	输入	原始输入	BF16、FLOAT16、FLOAT、UINT8、INT8、UINT16、INT16、UINT、INT、INT64、UINT64、BOOL、DOUBLE、COMPLEX64、COMPLEX32	ND
indices	输入	索引值	INT64、INT	ND
out	输出	索引之后的结果	BF16、FLOAT16、FLOAT、UINT8、INT8、UINT16、INT16、UINT、INT、INT64、UINT64、BOOL、DOUBLE、COMPLEX64、COMPLEX32	ND

约束说明

无

调用说明

调用方式	样例代码	说明
aclnn	test_aclnn_embedding	通过aclnn接口方式调用Embedding算子。