Description: Retrieves a set of values from weight based on indices, then aggregates them using the specified mode (max, sum, or mean) according to the offsets defined by offsets. Other parameters provide finer control over the computation process.
Shape derivation: Assume:
[object Object]When mode is sum:
[object Object]When mode is mean:
[object Object]When mode is max:
[object Object]
Each operator has calls. First, aclnnEmbeddingBagGetWorkspaceSize is called to obtain the workspace size required for computation and the executor that contains the operator computation flow. Then, aclnnEmbeddingBag is called to perform computation.
Parameters
[object Object]- [object Object]Atlas training series products[object Object] and [object Object]Atlas inference series products[object Object]: The data type cannot be BFLOAT16.
Returns
aclnnStatus: status code. For details, see .
The first-phase API implements input parameter verification. The following errors may be thrown.
[object Object]
- Deterministic computation:
- aclnnEmbeddingBag defaults to a deterministic implementation.
[object Object] and [object Object] support only [object Object].
The following example is for reference only. For details, see .