FilesLast commitLast update
12 days ago
1 month ago
2 months ago
6 days ago
9 months ago
12 hours ago
4 days ago
1 month ago
3 days ago
4 days ago
1 month ago
5 days ago
README

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.