Flatten

产品支持情况

产品	是否支持
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	输入张量	输入张量，支持非连续的Tensor，2D~8D。	INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64、BOOL、BFLOAT16、FLOAT、FLOAT16	ND
axis	输入属性	flatten计算的基准轴，取值范围为[-self.dim(), self.dim()-1]，目前仅支持axis = 1。	INT64	-
out	输出张量	输出2D张量，支持非连续的Tensor。	INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64、BOOL、BFLOAT16、FLOAT、FLOAT16	ND

约束说明

• 输入张量的维度不能大于8。
• axis必须在有效的维度范围内。
• 输入和输出的数据类型必须一致。

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_flatten	通过aclnnFlatten接口方式调用Flatten算子。