Flatten

功能

将tensor展平。

输入

一个输入：

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

输出

具有输入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