文件最后提交记录最后更新时间
23 天前
23 天前
1 个月前
16 天前
23 天前
23 天前
1 个月前
16 天前
1 个月前
23 天前
README

Pack

产品支持情况

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

功能说明

  • 算子功能:通过沿轴维度打包,将值中的张量列表打包成一个比值中的每个张量高一维度的张量。给定形状为(A,B,C)的张量的长度为N的列表;如果轴==0,则输出张量将具有形状(N,A,B,C)。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 需要级联的tensor列表。 COMPLEX128、COMPLEX64、BOOL、DOUBLE、FLOAT32、FLOAT16、BFLOAT16、INT16、INT32、INT64、INT8、UINT8、UINT16、UINT32、UINT64、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN ND
axis 可选属性 指定沿其打包的维度。 INT -
N 可选属性 指定要打包的tensor个数。 INT -
y 输出 输出结果。 COMPLEX128、COMPLEX64、BOOL、DOUBLE、FLOAT32、FLOAT16、BFLOAT16、INT16、INT32、INT64、INT8、UINT8、UINT16、UINT32、UINT64、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN ND
  • Atlas训练系列产品、Atlas推理系列产品、Atlas 200I/500 A2推理产品:不支持BFLOAT16、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN。
  • Atlas A2系列产品、Atlas A3系列产品:不支持HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN。

约束说明

无。

调用说明

调用方式 样例代码 说明
aclnn接口 test_aclnn_stack 通过aclnnStack接口方式调用Pack算子。