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

spelling: array

Browse Source 
Signed-off-by: Josh Soref <2119212+jsoref@users.noreply.github.com>
This commit is contained in:
Josh Soref 2024-01-01 23:03:10 -05:00
parent 7057f13b5f
commit 27f640e7c9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/ops_tests.cpp
View File

@ -1411,7 +1411,7 @@ TEST_CASE("test broadcast") {
x.eval(); x.eval();
CHECK_EQ(x.strides(), std::vector<size_t>{0, 0, 1}); CHECK_EQ(x.strides(), std::vector<size_t>{0, 0, 1});
// Broadcast on transposed arrray works // Broadcast on transposed array works
x = array({0, 1, 2, 3, 4, 5}, {2, 3}); x = array({0, 1, 2, 3, 4, 5}, {2, 3});
x = broadcast_to(transpose(x), {2, 3, 2}); x = broadcast_to(transpose(x), {2, 3, 2});
CHECK_EQ(x.shape(), std::vector<int>{2, 3, 2}); CHECK_EQ(x.shape(), std::vector<int>{2, 3, 2});