API: Implement `where` and `nonzero`

finch-tensor / finch-tensor-python

Sparse and Structured Tensor Programming in Python

MIT License

8 stars 3 forks source link

Closed mtsokol closed 6 months ago

mtsokol commented 6 months ago

Hi @willow-ahrens @hameerabbasi,

This PR adds Array API compatible where and nonzero functions.