FilesLast commitLast update
8 months ago
9 months ago
9 months ago
9 months ago
1 month ago
5 days ago
README

ConcatDV2

English | 简体中文

Product Support Status

Product Supported
Ascend 950PR&950DT products x
Atlas A3 products √
Atlas A2 products √
Atlas 200I/500 A2 inference products x
Atlas inference products x
Atlas training products x

Function Description

  • Operator function: concatenates multiple input tensors along the specified dimension, and outputs a tensor containing all input data concatenated in order.
  • Computation process:
    • Input:

      • Tensor list x[0], x[1],..., x[N-1]
      • Concatenation dimension concat_dim
    • Process:

      1. Check that all input tensors have the same data type;
      2. Check that all dimensions except concat_dim are identical;
      3. Concatenate along the concat_dim dimension in sequence: y = ConcatDV2(x[0], x[1], ..., x[N-1], axis = concat_dim)
    • Output: concatenated tensor y

Parameter Description

Parameter Name Input/Output/Attribute Description Data Type Data Format
x Input Dynamic input list, which is the input x[i] in the process diagram. BFLOAT16, FLOAT16, FLOAT, DOUBLE, INT32, UINT8, INT16, INT8, COMPLEX64, INT64, BOOL ND
concat_dim Input Specifies the concatenation dimension, which is concat_dim in the computation process. INT32, INT64 ND
y Output Output y in the computation process. BFLOAT16, FLOAT16, FLOAT, DOUBLE, INT32, UINT8, INT16, INT8, COMPLEX64, INT64, BOOL ND

Constraint Description

  • All input tensors must have the same shape except for the concatenation dimension.
  • The concatenation dimension concat_dim currently supports only 0.
  • All tensors in x must have the same data type.

Invocation Description

Invocation Method Sample Code Description
aclnn API test_aclnn_cat Invokes the ConcatDV2 operator through the aclnnCat API.