FilesLast commitLast update
22 hours ago
7 months ago
19 days ago
7 months ago
18 days ago
20 days ago
2 months ago
7 months ago
1 month ago
6 days ago
README

ChunkCat

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

Splits all tensors in tensors into num_chunks chunks along dimension dim, concatenates them along the dimension after dim, and finally converts the result to the data type of out.

Parameter Description

Parameter Name Input/Output/Attribute Description Data Type Data Format
x Input List of input tensors. FLOAT, FLOAT16, BFLOAT16 ND
dim Attribute Dimension along which tensors are split into chunks. INT -
num_chunks Attribute Number of chunks to split into. INT -
y Output Output tensor. FLOAT, FLOAT16, BFLOAT16 ND

Constraint Description

  • The data type or data format of elements in the x list is not within the supported range.
  • The data types of elements in the x list are inconsistent.
  • The shapes of tensors in the x list are not within 1 to 8 dimensions.
  • dim is not 0.

Invocation Description

Invocation Method Sample Code Description
aclnn API test_aclnn_chunk_cat Invokes the ChunkCat operator through the aclnnChunkCat API.