Compress

Description

Selects a slice from the input tensor along a given axis. The condition calculation result of each axis index is True.

Input

• input: tensor of type uint8, uint16, uint32, uint64, int8, int16, int32, int64, float16 or float, for rank r>=1.
• condition: 1D tensor of type bool, specifying the slice and the elements to be selected.

Attribute

axis (optional): int, axis for slicing. If axis is not provided, the input is flattened before the element is selected. The value range is [-r, r–1], where r is the rank of the input.

Output

output: output tensor. If axis is specified, it is a rank-r tensor; otherwise, it is a 1D tensor. The data type is the same as that of the input.

Constraints

None.

ONNX Opset Support

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