SplitV
English | 简体中文
Product Support Status
| Product | Supported |
|---|---|
| Ascend 950PR&950DT products | √ |
| Atlas A3 products | √ |
| Atlas A2 products | √ |
| Atlas 200I/500 A2 inference products | √ |
| Atlas inference products | √ |
| Atlas training products | √ |
Function Description
- Operator function: splits a tensor into num_split smaller tensors along the dimension split_dim according to size_splits.
Parameter Description
| Parameter Name | Input/Output/Attribute | Description | Data Type | Data Format |
|---|---|---|---|---|
| x | Input | List of tensors to be split. | FLOAT16, FLOAT32, DOUBLE, INT64, INT32, UINT8, UINT16, UINT32, UINT64, INT8, INT16, BOOL, COMPLEX64, COMPLEX128 | ND |
| size_splits | Input | Specifies a list that contains the size of each output tensor along the split dimension. | INT32, INT64 | ND |
| split_dim | Input | Dimension along which the tensor is split. | INT32, INT64 | ND |
| num_split | Attribute | Number of tensors to be split. | INT | - |
| y | Output | Output result. | FLOAT16, FLOAT32, DOUBLE, INT64, INT32, UINT8, UINT16, UINT32, UINT64, INT8, INT16, BOOL, COMPLEX64, COMPLEX128 | ND |
- Atlas training products, Atlas inference products, Atlas 200I/500 A2 inference products, and Atlas 200/300/500 inference products: BFLOAT16 is not supported.
Constraint Description
- Each element in size_splits is greater than or equal to 1.
- The length of size_splits equals the value of num_split.
- The sum of the elements in size_splits equals the size of the dimension split_dim.
Invocation Description
| Invocation Method | Sample Code | Description |
|---|---|---|
| aclnn API | test_aclnn_split_tensor | Invokes the SplitV operator through the aclnnSplitTensor API. |
| aclnn API | test_aclnn_split_with_size | Invokes the SplitV operator through the aclnnSplitWithSize API. |