ChunkCat

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	×
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	×
Atlas 训练系列产品	×

功能说明

将tensors中所有tensor先按照维度dim切分为num_chunks块，再按照dim后一维进行级联，最后转换为out的数据类型。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	输入tensor列表。	FLOAT、FLOAT16、BFLOAT16	ND
dim	属性	需要切分块的维度。	INT	-
num_chunks	属性	指定要切分块的数量。	INT	-
y	输出	输出tensor。	FLOAT、FLOAT16、BFLOAT16	ND

约束说明

• x列表中元素的数据类型和数据格式不在支持的范围之内。
• x列表中元素的数据类型不一致。
• x列表中tensor的shape不在1~8维。
• dim不为0。

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_chunk_cat	通过aclnnChunkCat接口方式调用ChunkCat算子。