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