zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-18 10:26:56 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

MLX in C++ example (#1736)

Browse Source 
* MLX in C++ example

* nits

* fix docs
This commit is contained in:
Awni Hannun
2025-01-02 19:09:04 -08:00
committed by GitHub
parent 8544b42007
commit c9d30aa6ac
13 changed files with 242 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/ops.h
View File

@@ -914,7 +914,7 @@ inline array gather(
return gather(a, {indices}, std::vector<int>{axis}, slice_sizes, s);
}
/** Returns Kronecker Producct given two input arrays. */
/** Compute the Kronecker product of two arrays. */
array kron(const array& a, const array& b, StreamOrDevice s = {});
/** Take array slices at the given indices of the specified axis. */