GatherElementsV2

产品支持情况

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

功能说明

• 算子功能：对输入tensor中指定的维度dim进行数据聚集。

• 计算公式： 给定张量$x$，维度$d$，和一个索引张量$index$，定义$n$是$x$的维度，$i_d$表示维度$d$的索引，$inde{x}_{i_d}$表示索引张量$index$在维度$d$上的第$i_d$个索引值。对指定维度d的gather功能可以用如下的数学公式表示：

  $$gather(X,index,d{)}_{i_0,i_1,\cdots ,i_{d-1},i_{d+1},\cdots ,i_{n-1}}=x_{i_0,i_1,\cdots ,i_{d-1},inde{x}_{i_d},i_{d+1},\cdots ,i_{n-1}}$$

• 示例：

  • 示例1： 假设输入张量$x=\left[\begin{array}{ccc}1& 2& 3\\ 4& 5& 6\\ 7& 8& 9\end{array}\right]$，索引张量$index=\left[\begin{array}{cc}0& 2\\ 1& 0\end{array}\right]$，$dim=0$，那么输出张量$y=\left[\begin{array}{cc}1& 8\\ 4& 2\end{array}\right]$，具体计算过程如下：

    $$\begin{array}{rl}{y}_{0,0}& =x_{inde{x}_{0,0},0}=x_{0,0}=1\\ y_{0,1}& =x_{inde{x}_{0,1},1}=x_{2,1}=8\\ y_{1,0}& =x_{inde{x}_{1,0},0}=x_{1,0}=4\\ y_{1,1}& =x_{inde{x}_{1,1},1}=x_{0,1}=2\end{array}$$

  • 示例2： 假设输入张量$x=\left[\begin{array}{ccc}1& 2& 3\\ 4& 5& 6\\ 7& 8& 9\end{array}\right]$，索引张量$index=\left[\begin{array}{cc}0& 2\\ 1& 0\end{array}\right]$，$dim=1$，那么输出张量$y=\left[\begin{array}{cc}1& 3\\ 5& 4\end{array}\right]$，具体计算过程如下：

    $$\begin{array}{rl}{y}_{0,0}& =x_{0,inde{x}_{0,0}}=x_{0,0}=1\\ y_{0,1}& =x_{0,inde{x}_{0,1}}=x_{0,2}=3\\ y_{1,0}& =x_{1,inde{x}_{1,0}}=x_{1,1}=5\\ y_{1,1}& =x_{1,inde{x}_{1,1}}=x_{1,0}=4\end{array}$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	公式中的x。	FLOAT、FLOAT16、BFLOAT16、INT32	ND
index	输入	公式中的index。	INT32	ND
y	输出	公式中的y。	FLOAT、FLOAT16、BFLOAT16、INT32	ND
dim	可选属性	公式中的d。 默认值为0。	Int	-

• Kirin X90/Kirin 9030 处理器系列产品：x和y的数据类型不支持BFLOAT16。

约束说明

无

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_gather_elements_v2	通过aclnnGather接口方式调用GatherElementsV2算子。