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

Scatter vjp (#394)

Browse Source 
* Add a first scatter vjp
* Implement the scatter_add vjp
* Add array.at to implement user friendly scatters
This commit is contained in:
Angelos Katharopoulos
2024-01-09 13:36:51 -08:00
committed by GitHub
parent e9ca65c939
commit 961435a243
7 changed files with 360 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/src/ops.cpp
View File

@@ -3310,7 +3310,6 @@ void init_ops(py::module_& m) {
Returns:
result (array): The tensor dot product.
)pbdoc");
m.def(
"inner",
&inner,
@@ -3331,7 +3330,6 @@ void init_ops(py::module_& m) {
Returns:
result (array): The inner product.
)pbdoc");
m.def(
"outer",
&outer,