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

Add view op (#1179)

Browse Source 
* add view primitive

* nit

* fix view
This commit is contained in:
Awni Hannun
2024-06-04 08:05:27 -07:00
committed by GitHub
parent 81def6ac76
commit ea9090bbc4
14 changed files with 202 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/ops_tests.cpp
View File

@@ -3473,3 +3473,15 @@ TEST_CASE("test trace") {
auto out4 = trace(in, 0, 1, 2, float32);
CHECK(array_equal(out4, array({3, 11}, {2})).item<bool>());
}
TEST_CASE("test view") {
auto in = array(3);
CHECK_THROWS(view(in, int64));
in = array({1, 2, 3});
CHECK_THROWS(view(in, int64));
in = array({1, 2, 3, 4}, int64);
auto out = view(in, int32);
CHECK(array_equal(out, array({1, 0, 2, 0, 3, 0, 4, 0})).item<bool>());
}