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