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

spelling: axes

Browse Source 
Signed-off-by: Josh Soref <2119212+jsoref@users.noreply.github.com>
This commit is contained in:
Josh Soref 2024-01-01 22:49:20 -05:00
parent f28bcda866
commit 9ad62ea6a0
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mlx/ops.cpp
View File

@ -350,7 +350,7 @@ array squeeze(
ax = ax < 0 ? ax + a.ndim() : ax; ax = ax < 0 ? ax + a.ndim() : ax;
if (ax < 0 || ax >= a.ndim()) { if (ax < 0 || ax >= a.ndim()) {
std::ostringstream msg; std::ostringstream msg;
msg << "[squeeze] Invalid axies " << ax << " for array with " << a.ndim() msg << "[squeeze] Invalid axes " << ax << " for array with " << a.ndim()
<< " dimensions."; << " dimensions.";
throw std::invalid_argument(msg.str()); throw std::invalid_argument(msg.str());
} }
@ -405,7 +405,7 @@ array expand_dims(
ax = ax < 0 ? ax + out_ndim : ax; ax = ax < 0 ? ax + out_ndim : ax;
if (ax < 0 || ax >= out_ndim) { if (ax < 0 || ax >= out_ndim) {
std::ostringstream msg; std::ostringstream msg;
msg << "[squeeze] Invalid axies " << ax << " for output array with " msg << "[squeeze] Invalid axes " << ax << " for output array with "
<< a.ndim() << " dimensions."; << a.ndim() << " dimensions.";
throw std::invalid_argument(msg.str()); throw std::invalid_argument(msg.str());
} }