Flatten and unflatten (#1692)

* flatten and unflatten * fix grad * fix shape infer * use squeeze + unsqueeze in get_item
2025-09-26 15:58:14 +08:00 · 2024-12-11 21:51:37 -08:00
parent 0bf19037ca
commit 4e1e9520e1
19 changed files with 363 additions and 93 deletions
--- a/docs/src/python/ops.rst
+++ b/docs/src/python/ops.rst
@@ -168,6 +168,7 @@ Operations
   tri
   tril
   triu
+   unflatten
   var
   view
   where