Skip to content

Implement dpctl.tensor.where #1120

Closed
@oleksandr-pavlyk

Description

@oleksandr-pavlyk

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.

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions