Flatten

功能

将输入张量input展平为 2D 矩阵。如果输入张量的形状为(d_0, d_1, ...d_n),则输出的形状为(d_0 * d_1 ...d_axis-1, d_axis * ...* dn)。

输入

input:输入Tensor,秩r >2,数据类型:float16、float。

属性

axis:数据类型支持int(默认为1)。指定一个输入维度(不包含)平铺到输出的外部维度。axis的值必须在[-r, r]范围内,其中r是输入Tensor的秩。负值表示从后面开始计算尺寸。当轴=0时,输出Tensor的形状为(1, (d_0 * d_1...d_n)),其中输入Tensor的形状为(d_0, d_1, ...d_n)。

输出

output:输出Tensor,具有输入tensor的内容的2D Tensor。

约束与限制

无。

支持的ONNX版本

Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18