[object Object]

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

Description: Selects elements in the input tensor ([object Object]) as the output according to the Boolean mask tensor ([object Object]) and returns a new one-dimensional tensor.

[object Object]

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

• [object Object]
• [object Object]

[object Object]

• Parameters:

  • [object Object] (aclTensor*, compute input): input tensor [object Object] in the function, aclTensor on the device. The shape must meet the with [object Object]. are supported. The supports ND.
    • [object Object]Atlas training products[object Object] and [object Object]Atlas inference products[object Object]: The data type can be FLOAT16, FLOAT32, DOUBLE, INT8, INT16, INT32, INT64, UINT8, or BOOL.
    • [object Object]Atlas A2 training products/Atlas A2 inference products[object Object] and [object Object]Atlas A3 training products/Atlas A3 inference products[object Object]: The data type can be BFLOAT16, FLOAT16, FLOAT32, DOUBLE, INT8, INT16, INT32, INT64, UINT8, or BOOL.
  • [object Object] (aclTensor*, compute input): Boolean mask tensor [object Object] in the function, aclTensor on the device. The shape must meet the with [object Object]. are supported. The supports ND. The data type can be UINT8 or BOOL.
  • [object Object] (aclTensor*, compute output): output one-dimensional tensor in the function, aclTensor on the device. The shape is one-dimensional, and the number of elements is shapeSize after broadcasting is performed between [object Object] and [object Object]. are not supported. The supports ND.
    • [object Object]Atlas training products[object Object] and [object Object]Atlas inference products[object Object]: The data type can be FLOAT16, FLOAT32, DOUBLE, INT8, INT16, INT32, INT64, UINT8, or BOOL.
    • [object Object]Atlas A2 training products/Atlas A2 inference products[object Object] and [object Object]Atlas A3 training products/Atlas A3 inference products[object Object]: The data type can be BFLOAT16, FLOAT16, FLOAT32, DOUBLE, INT8, INT16, INT32, INT64, UINT8, or BOOL.
  • [object Object] (uint64_t*, output): size of the workspace to be allocated on the device.
  • [object Object] (aclOpExecutor**, output): operator executor, containing the operator computation process.

• Returns:

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

[object Object]

• Parameters:

  • [object Object] (void*, input): memory address of the workspace to be allocated on the device.

  • [object Object] (uint64_t, input): size of the workspace to be allocated on the device, which is obtained by the first-phase API [object Object].

  • [object Object] (aclOpExecutor*, input): operator executor, containing the operator computation process.

  • [object Object] (aclrtStream, input): stream for executing the task.

• Returns:

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

[object Object]

• Deterministic computation
  • [object Object] defaults to deterministic implementation.

The function is supported in the broadcast scenario, but the performance is not guaranteed. In the broadcast scenario, the BroadcastTo operator is inserted to solve the problem.

[object Object]

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