mlx/python
Jacket 3f7aba8498
Implement diagonal operator (#562)
* Implement diagonal operator

This implements mx.diagonal in operator level, inspired by
@ManishAradwad.

* added `mx.diag` with tests

* corrected few things

* nits in bindings

* updates to diag

---------

Co-authored-by: ManishAradwad <manisharadwad@gmail.com>
Co-authored-by: Awni Hannun <awni@apple.com>
2024-01-30 09:45:48 -08:00
..
mlx Add more checks and clearer error messages to conv operations (#563) 2024-01-26 15:13:26 -08:00
src Implement diagonal operator (#562) 2024-01-30 09:45:48 -08:00
tests Implement diagonal operator (#562) 2024-01-30 09:45:48 -08:00