Flatten
输入
x:
- 是否必填: 必填
- 数据类型: float16,float32,int8,uint8,int16,uint16,int32,uint32,int64,uint64
- 参数解释: 输入Tensor
- 规格限制: 无
属性
- axis:
- 是否必填: 非必填
- 数据类型: int
- 参数解释: 需要flatten的开始轴,可以为负数,如果为-1表示最后一个轴,默认值为1
- 规格限制: [-rank(x), rank(x))
- end_axis:
- 是否必填: 非必填
- 数据类型: int
- 参数解释: 需要flatten的结束轴,可以为负数,如果为-1表示最后一个轴,默认值1
- 规格限制: [0, rank(x))
输出
y:
- 是否必填: 必填
- 数据类型: float16,float32,int8,uint8,int16,uint16,int32,uint32,int64,uint64
- 参数解释: 输出tensor
- 规格限制: 无
父主题: 支持Caffe算子清单