implemented Flatten Module (#149)

* implemented flatten op --------- Co-authored-by: Awni Hannun <[email protected]>
2023-12-16 21:54:37 -08:00
parent eebd7c275d
commit 52e1589a52
8 changed files with 113 additions and 2 deletions
@@ -41,6 +41,7 @@ Operations
   expand_dims
   eye
   floor
+   flatten
   full
   greater
   greater_equal