
Pack【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√Atlas 200I/500 A2 推理产品√Atlas 推理系列产品√Atlas 训练系列产品√功能说明算子功能通过沿轴维度打包将值中的张量列表打包成一个比值中的每个张量高一维度的张量。给定形状为ABC的张量的长度为N的列表如果轴0则输出张量将具有形状NABC。参数说明参数名输入/输出/属性描述数据类型数据格式x输入需要级联的tensor列表。COMPLEX128、COMPLEX64、BOOL、DOUBLE、FLOAT32、FLOAT16、BFLOAT16、INT16、INT32、INT64、INT8、UINT8、UINT16、UINT32、UINT64NDaxis可选属性指定沿其打包的维度。INT-N可选属性指定要打包的tensor个数。INT-y输出输出结果。COMPLEX128、COMPLEX64、BOOL、DOUBLE、FLOAT32、FLOAT16、BFLOAT16、INT16、INT32、INT64、INT8、UINT8、UINT16、UINT32、UINT64NDAtlas 训练系列产品、Atlas 推理系列产品不支持BFLOAT16。约束说明无。调用说明调用方式样例代码说明aclnn接口test_aclnn_stack通过aclnnStack接口方式调用Pack算子。【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考