Where

功能

根据条件从两个输入中选择元素。

输入

三个输入：

condition，条件，数据类型：bool。

x：一个tensor，条件为true时从x中选取元素，数据类型支持float16、float、int8、int32、uint8。

y：一个tensor，条件为false时从y中选取元素，和x的数据类型一致。

输出

一个输出：

一个tensor，数据类型和输入x一致。

支持的ONNX版本

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