added tri / tril / triu (#170)

* added tri / tril / triu * fixed tests * ctest tests * tri overload and simplified tests * changes from comment * more tests for m * ensure assert if not 2-D * remove broadcast_to * minor tweaks --------- Co-authored-by: Awni Hannun <awni@apple.com>
2025-09-01 04:24:36 +08:00 · 2023-12-15 20:30:34 -05:00
parent 2e02acdc83
commit dc2edc762c
9 changed files with 207 additions and 12 deletions
--- a/docs/src/python/ops.rst
+++ b/docs/src/python/ops.rst
@@ -95,6 +95,9 @@ Operations
   tan
   tanh
   transpose
+   tri
+   tril
+   triu
   var
   where
   zeros