Compress
功能
按指定轴进行切片。
输入
两个输入:
input:维度大于等于1的tensor,支持类型:uint8、uint16、uint32、uint64、int8、int16、int32、int64、float16、float。
condition:1维tensor,用于指定切片和需要选择的元素,支持类型:bool。
输出
一个输出:
output:tensor,类型:与输入input一致。
属性
axis:可选,int类型,进行切片的轴,如果没有指定轴,在切片之前将输入tensor展平。取值范围是[-r,r-1],r为输入input的维数。
支持的ONNX版本
Opset v9/v10/v11/v12/v13/v14/v15/v16/v17/v18
父主题: 支持ONNX算子清单