[object Object]

[object Object][object Object]undefined[object Object]

• Description: Gathers data of a specific dimension dim in the input tensor.

• Formula: Assume that $self$ indicates a tensor, $d$ indicates a dimension, and $index$ indicates an index tensor. If $n$ is defined as the dimension of $self$, $i_d$ is the index of $d$, and $index_{i_d}$ indicates the $i_d$th index value of tensor $index$ in dimension $d$. To gather data of the specific dimension d, use the following formula:

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

• Example:

  • Example 1: Assume that the input tensor self = $\begin{bmatrix}1 & 2 & 3\\ 4 & 5 & 6\\ 7 & 8 & 9\end{bmatrix}$ and the index tensor index = $\begin{bmatrix}0 & 2\\ 1 & 0\end{bmatrix}$ with dim = 0. Then, the output tensor out = $\begin{bmatrix}1 & 8\\ 4 & 2\end{bmatrix}$. The calculation process is as follows:

    $$\begin{aligned} out_{0,0}&=self_{index_{0,0}, 0}=self_{0,0}=1 \\ out_{0,1}&=self_{index_{0,1}, 1}=self_{2,1}=8 \\ out_{1,0}&=self_{index_{1,0}, 0}=self_{1,0}=4 \\ out_{1,1}&=self_{index_{1,1}, 1}=self_{0,1}=2 \end{aligned}$$

  • Example 2: Assume that the input tensor self = $\begin{bmatrix}1 & 2 & 3\\ 4 & 5 & 6\\ 7 & 8 & 9\end{bmatrix}$ and the index tensor index = $\begin{bmatrix}0 & 2\\ 1 & 0\end{bmatrix}$ with dim = 1. Then, the output tensor out = $\begin{bmatrix}1 & 3\\ 5 & 4\end{bmatrix}$. The calculation process is as follows:

    $$\begin{aligned} out_{0,0}&=self_{0, index_{0,0}}=self_{0,0}=1 \\ out_{0,1}&=self_{0, index_{0,1}}=self_{0,2}=3 \\ out_{1,0}&=self_{1, index_{1,0}}=self_{1,1}=5 \\ out_{1,1}&=self_{1, index_{1,1}}=self_{1,0}=4 \end{aligned}$$

[object Object]

Each operator has calls. First, aclnnGatherGetWorkspaceSize is called to obtain the workspace size required for computation and the executor that contains the operator computation process. Then, aclnnGather is called to perform computation.

[object Object]

• Parameters

  [object Object]

• Returns

  [object Object]: status code. For details, see .

  The first-phase API implements input parameter verification. The following errors may be thrown.

  [object Object]

[object Object]

• Parameters

  [object Object]

• Returns

  [object Object]: status code. For details, see .

[object Object]

• Deterministic computation:
  • aclnnGather defaults to a deterministic implementation.

[object Object]

The following example is for reference only. For details, see .