[object Object]

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

Description: Splits the input self along the [object Object] axis into the size of each element in [object Object].

[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*, computation input): input tensor to be split. are supported. The can be ND.

      • [object Object]Atlas training products[object Object]: The data type can be FLOAT, FLOAT16, DOUBLE, INT32, INT64, INT16, INT8, UINT8, BOOL, COMPLEX128, or COMPLEX64. When the number of outputs is greater than 32, DOUBLE, COMPLEX128, or COMPLEX64 is not supported.
      • [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 FLOAT, FLOAT16, DOUBLE, INT32, INT64, INT16, INT8, UINT8, BOOL, COMPLEX128, COMPLEX64, or BFLOAT16. When the number of outputs is greater than 32, DOUBLE, COMPLEX128, or COMPLEX64 is not supported.
    • [object Object] (aclIntArray*, computation input): size of each split block. The data type can be INT64 or INT32. The total size of all blocks must be equal to the shape size of [object Object] in the [object Object] dimension.

    • [object Object] (int64_t, computation input): dimension along which the input tensor is split. The data type can be INT64.

    • [object Object] (aclTensorList*, computation output): output tensor list after splitting. are supported. The can be ND.

      • [object Object]Atlas training products[object Object]: The data type can be FLOAT, FLOAT16, DOUBLE, INT32, INT64, INT16, INT8, UINT8, BOOL, COMPLEX128, or COMPLEX64. When the number of outputs is greater than 32, DOUBLE, COMPLEX128, or COMPLEX64 is not supported.
      • [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 FLOAT, FLOAT16, DOUBLE, INT32, INT64, INT16, INT8, UINT8, BOOL, COMPLEX128, COMPLEX64, or BFLOAT16. When the number of outputs is greater than 32, DOUBLE, COMPLEX128, or COMPLEX64 is not supported.
    • [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]
[object Object]
  • Parameters:

    • [object Object] (void*, input): 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 calling 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 compute:
    • [object Object] defaults to a deterministic implementation.
[object Object]

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

[object Object]