Where
Description
Selects elements from two inputs based on conditions.
Input
Three inputs:
condition: condition of type bool.
x: tensor of type float16, float, int8, int32, or uint8. Selects elements from x when the condition is true.
y: tensor. When the condition is false, elements are selected from y. Has the identical data type as x.
Output
One output:
Tensor of the identical data type as input x.
ONNX Opset Support
Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18
Parent topic: Operators Specifications of the AI Framework