[object Object]

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

• Description: Extracts elements from the specified dimension dim of the input tensor based on the index number in index. batchDims indicates the batch size. The result is saved to the out tensor.

• Example:

  For example, when batchDims is 0, the input tensor self = \begin{bmatrix}1 & 2 & 3 \ 4 & 5 & 6 \ 7 & 8 & 9\end{bmatrix}$ and the index tensor index = [1, 0].

  • Result of dim = 0: $out = \begin{bmatrix}4 & 5 & 6 \\ 1 & 2 & 3\end{bmatrix}$

  • Result of dim = 1: $out = \begin{bmatrix}2 & 1\\ 5 & 4\\ 8 & 7\end{bmatrix}$

  The computation process is as follows: Take a 3D tensor with shape (3, 2, 2) as an example: Assume self = $\begin{bmatrix}[[1,&2],&[3,&4]], \\ [[5,&6],&[7,&8]], \\ [[9,&10],&[11,&12]]\end{bmatrix}$ with index = [1, 0], the indices corresponding to dim 0, 1, and 2 are $l$, $m$, and $n$, respectively. index is 1D (if it is 0D, it is considered as a 1D tensor with the size of 1).

  • dim = 0: I = index[i];    out$[i][m][n]$ = self$[I][m][n]$

  • dim = 1: J = index[j];    out$[l][j][n]$ = self$[l][J][n]$

  • dim = 2: K = index[k];    out$[l][m][k]$ = self$[l][m][K]$

  When batchDims is 1, take a 4D tensor as an example. The tensor self with shape (3, 3, 2, 2) and the tensor index with shape (3, 2) can be obtained from performing the gather operation three times with batchDims being 0 and dim being dim – batchDims.

[object Object]

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

[object Object]

• Parameters

  [object Object]
  • [object Object]Atlas A2 training products/Atlas A2 inference products[object Object], [object Object]Atlas A3 training products/Atlas A3 inference products[object Object]: Currently, dim supports only 0 and batchDims supports only 0.

• 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:
  • aclnnGatherV3 defaults to a deterministic implementation.

[object Object]

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