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

Add mx.meshgrid (#961)

Browse Source
This commit is contained in:
Abe Leininger
2024-04-09 14:43:08 -04:00
committed by GitHub
parent ae812350f9
commit a1a31eed27
6 changed files with 161 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
mlx/ops.h
View File

@@ -197,6 +197,13 @@ std::vector<array> split(
std::vector<array>
split(const array& a, const std::vector<int>& indices, StreamOrDevice s = {});
/** A vector of coordinate arrays from coordinate vectors. */
std::vector<array> meshgrid(
const std::vector<array>& arrays,
bool sparse = false,
std::string indexing = "xy",
StreamOrDevice s = {});
/**
* Clip (limit) the values in an array.
*/