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

Added linspace (#181)

Browse Source 
* linspace ops support

---------

Co-authored-by: Awni Hannun <awni@apple.com>
This commit is contained in:
Abe Leininger
2023-12-18 19:57:55 -08:00
committed by GitHub
parent f4f6e17d45
commit e6872a4149
6 changed files with 92 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/ops_tests.cpp
View File

@@ -2201,3 +2201,17 @@ TEST_CASE("test clipping with only max") {
auto clipped = clip(a, std::nullopt, array(4.0f));
CHECK(array_equal(clipped, expected).item<bool>());
}
TEST_CASE("test linspace") {
auto x = linspace(0, 10, 5);
auto expected = array({0.0f, 2.5f, 5.0f, 7.5f, 10.0f}, {5});
CHECK(array_equal(x, expected).item<bool>());
x = linspace(0, 10, 5, int32);
expected = array({0, 2, 5, 7, 10}, {5});
CHECK(array_equal(x, expected).item<bool>());
x = linspace(0, 1, 0);
expected = array(std::initializer_list<float>{}, {0});
CHECK(array_equal(x, expected).item<bool>());
}