Flatten【免费下载链接】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 训练系列产品√功能说明算子功能将输入Tensor基于给定的axis扁平化为一个2D的Tensor。计算公式若输入self的shape为(d₀, d₁, ..., dₙ)则输出out的shape为(d₀×d₁×...×dₐₓᵢₛ₋₁, dₐₓᵢₛ×dₐₓᵢₛ₊₁×...×dₙ)若axis取值为0则输出out的shape为(1, d₀×d₁×...×dₙ)参数说明参数名输入/输出/属性描述数据类型数据格式self输入张量输入张量支持非连续的Tensor2D~8D。INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64、BOOL、BFLOAT16、FLOAT、FLOAT16NDaxis输入属性flatten计算的基准轴取值范围为[-self.dim(), self.dim()-1]目前仅支持axis 1。INT64-out输出张量输出2D张量支持非连续的Tensor。INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64、BOOL、BFLOAT16、FLOAT、FLOAT16ND约束说明输入张量的维度不能大于8。axis必须在有效的维度范围内。输入和输出的数据类型必须一致。调用说明调用方式样例代码说明aclnn接口test_aclnn_flatten通过aclnnFlatten接口方式调用Flatten算子。【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考