Compress

功能

沿给定轴从输入张量中选择切片，其中每个轴索引的condition计算结果为True。

输入

• input：秩 r >= 1 的Tensor，数据类型支持uint8、uint16、uint32、uint64、int8、int16、int32、int64、float16、float。
• condition：一维Tensor，用于指定切片和需要选择的元素，数据类型支持bool。

属性

axis（可选）：数据类型为int，进行切片的轴，如果未提供axis，则在选择元素之前将input展平。取值范围为[-r，r-1]，其中r为输入input的维数。

输出

output：输出Tensor，如果指定了axis，则为r秩的Tensor，否则是秩为1的Tensor。数据类型与输入input一致。

约束与限制

无。

支持的ONNX版本

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