Description: Copies the data from data to out, and updates the data in out based on updates at the specified indices.
Each operator has calls. First, aclnnScatterNdGetWorkspaceSize is called to obtain the workspace size required for computation and the executor that contains the operator computation process. Then, aclnnScatterNd is called to perform computation.
[object Object][object Object]
Parameters:
data (aclTensor*, compute input): aclTensor on the device. The data type is the same as that of updates and out. The shape meets the requirement: 1 <= rank(data) <= 8. are supported. The can be ND.
- [object Object]Atlas A2 training products/Atlas A2 inference products[object Object] and [object Object]Atlas A3 training series products/Atlas A3 inference series products[object Object]: The data type can be FLOAT16, FLOAT, BOOL, or BFLOAT16.
- [object Object]Atlas training series products[object Object] and [object Object]Atlas inference series products[object Object]: The data type can be FLOAT16, FLOAT, or BOOL.
indices (aclTensor*, compute input): aclTensor on the device. The data type can be INT32 or INT64. The following requirements must be met: indices.shape[-1] ≤ rank(data), 1 ≤ rank(indices) ≤ 8. are supported. The can be ND. Only non-negative indices are supported. The data in indices cannot be out-of-bounds.
updates (aclTensor*, compute input): aclTensor on the device. The data type is the same as that of data and out. The shape must meet the following requirements: rank(updates) = rank(data) + rank(indices) – indices.shape[-1] – 1, 1 ≤ rank(updates) ≤ 8. are supported. The can be ND.
- [object Object]Atlas A2 training products/Atlas A2 inference products[object Object] and [object Object]Atlas A3 training series products/Atlas A3 inference series products[object Object]: The data type can be FLOAT16, FLOAT, BOOL, or BFLOAT16.
- [object Object]Atlas training series products[object Object] and [object Object]Atlas inference series products[object Object]: The data type can be FLOAT16, FLOAT, or BOOL.
out (aclTensor*, compute output): aclTensor on the device. The data type is the same as that of data and out. The shape is the same as that of data. are supported. The can be ND.
- [object Object]Atlas A2 training products/Atlas A2 inference products[object Object] and [object Object]Atlas A3 training series products/Atlas A3 inference series products[object Object]: The data type can be FLOAT16, FLOAT, BOOL, or BFLOAT16.
- [object Object]Atlas training series products[object Object] and [object Object]Atlas inference series products[object Object]: The data type can be FLOAT16, FLOAT, or BOOL.
workspaceSize (uint64_t*, output): size of the workspace to be allocated on the device.
executor (aclOpExecutor**, output): operator executor, containing the operator computation process.
Returns:
Parameters:
- workspace (void *, input): address of the workspace to be allocated on the device.
- workspaceSize (uint64_t, input): size of the workspace to be allocated on the device, which is obtained by the first-phase API aclnnScatterNdGetWorkspaceSize.
- executor (aclOpExecutor *, input): operator executor, containing the operator computation process.
- stream (aclrtStream, input): stream for executing the task.
Returns:
- Deterministic compute:
- aclnnScatterNd defaults to a deterministic implementation.
The following example is for reference only. For details, see .