Description: Maps a set of indices to a vector space and thereby quantizes the data. The 2D weight tensor of the embedding operator is weight, which has m+1 rows and n columns. For any input index tensor indices (for example, 1 row and 3 columns), the output out is a tensor with 3 rows and n columns, as shown below.
Each operator has calls. First, aclnnEmbeddingGetWorkspaceSize is called to obtain the workspace size required for computation and the executor that contains the operator computation flow. Then, aclnnEmbedding is called to perform computation.
Parameters
[object Object]- [object Object]Atlas inference products[object Object], [object Object]Atlas training products[object Object], and [object Object]Atlas 200I/500 A2 inference products[object Object]: The data type cannot be BFLOAT16.
Returns
[object Object]: status code. For details, see .The first-phase API implements input parameter verification. The following errors may be thrown.
[object Object]
- Deterministic computation:
- aclnnEmbedding defaults to a deterministic implementation.
The following example is for reference only. For details, see .