zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-31 07:58:14 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Add inner / outer op (#348)

Browse Source 
* inner / outer impl

* python tests

* ops list and ack

* updated descriptions

* use test helper

* removed dtype check and flatten outer to 1-D

* updated docs

* just use the reshape to flatten
This commit is contained in:
Diogo
2024-01-07 12:01:09 -05:00
committed by GitHub
parent 6ea6b4258d
commit 449b43762e
7 changed files with 140 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/src/python/ops.rst
View File

@@ -49,6 +49,7 @@ Operations
greater
greater_equal
identity
inner
less
less_equal
linspace
@@ -71,6 +72,7 @@ Operations
negative
ones
ones_like
outer
partition
pad
prod