Where

功能

根据condition，从X或Y中返回相应的元素。

输入

condition：bool类型，若为True，返回X；否则，返回Y。

X：输入Tensor，数据类型支持float16、float、uint8、int8、int32。

Y：输入Tensor，数据类型与X保持一致。

输出

output：输出Tensor，数据类型与X和Y保持一致，shape为X和Y经过广播后的shape。

限制与约束

无。

支持的ONNX版本

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