add mx.trace (#1143) (#1147)

* working c++ trace implementation * updated throw + added overloads * added python binding for trace function * pre-commit reformatting * add trace to docs * resolve comments * remove to_stream call
2025-09-26 15:58:14 +08:00 · 2024-05-22 18:50:27 -04:00
parent e110ca11e2
commit 79ef49b2c2
6 changed files with 161 additions and 0 deletions
--- a/docs/src/python/ops.rst
+++ b/docs/src/python/ops.rst
@@ -150,6 +150,7 @@ Operations
   tensordot
   tile
   topk
+   trace
   transpose
   tri
   tril