implemented Flatten Module (#149)

* implemented flatten op --------- Co-authored-by: Awni Hannun <awni@apple.com>
2025-10-31 07:58:14 +08:00 · 2023-12-17 06:54:37 +01:00
parent eebd7c275d
commit 52e1589a52
8 changed files with 113 additions and 2 deletions
--- a/docs/src/python/ops.rst
+++ b/docs/src/python/ops.rst
@@ -41,6 +41,7 @@ Operations
   expand_dims
   eye
   floor
+   flatten
   full
   greater
   greater_equal