Flatten

功能

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

输入

一个输入：

input：多维tensor，数据类型：float16、float。

输出

output：具有输入tensor的内容的2Dtensor。

属性

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

支持的ONNX版本

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