[object Object]

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

Description: Splits the input [object Object] along the [object Object] axis into equally-sized sections based on the [object Object] size. If the [object Object] axis cannot be exactly divided, the size of a non-last block is equal to [object Object], and the size of the last block is less than [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. The [1, 8] dimensions are supported.

    • [object Object]Atlas inference products[object Object] and [object Object]Atlas training products[object Object]: The data type can be FLOAT, FLOAT16, DOUBLE, INT32, INT64, INT16, INT8, UINT8, BOOL, COMPLEX128, or COMPLEX64.
    • [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.

  • [object Object] (uint64_t, computation input): UINT64, which indicates the size of the block evenly split along the [object Object] axis. The last block can be smaller than [object Object].

  • [object Object] (int64_t, computation input): INT64, which indicates the dimension of the input tensor to be split. The value range is [–self.dim(), self.dim()).

  • [object Object] (aclTensorList*, computation output): list of output tensors after splitting. are supported. The can be ND. The data type of each output must be the same. The number of dimensions of each output tensor in [object Object] is the same as that of [object Object].

    • [object Object]Atlas inference products[object Object] and [object Object]Atlas training products[object Object]: The data type can be FLOAT, FLOAT16, DOUBLE, INT32, INT64, INT16, INT8, UINT8, BOOL, COMPLEX128, or COMPLEX64.
    • [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.

  • [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): 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 .