昇腾社区首页
中文
注册

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
  • 规格限制: 无