Closed
Description
Function dpctl.tensor.where
is required by array-API specificaiton (link).
dpctl.where(condition, x1, x2) -> array
returns elements chosen from x1
or x2
depending on condition
.
One needs implementation kernel only for condition
of boolean type, and same data type arrays x1
and x2
.