三目运算。
z = cond ? x : y;
输出z的维度为输入x与y广播后的结果。要求cond, x, y必须是可广播的。
struct WhereParam {};
参数 |
维度 |
数据类型 |
格式 |
|---|---|---|---|
cond |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
int8 |
ND |
x |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16 |
ND |
y |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16 |
ND |
参数 |
维度 |
数据类型 |
格式 |
|---|---|---|---|
z |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16 |
ND |