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

rebase

Browse Source
This commit is contained in:
CircleCI Docs
2024-12-06 21:22:39 +00:00
parent 8ba8a39e43
commit 04b749a588
896 changed files with 12293 additions and 11212 deletions
Download Patch File Download Diff File Expand all files Collapse all files

110
docs/build/html/ops_8h.html vendored
View File

@@ -136,35 +136,35 @@ Functions</h2></td></tr>
<tr class="memitem:ga0e58c24fc5668e5a521e5b45e8370a62" id="r_ga0e58c24fc5668e5a521e5b45e8370a62"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga0e58c24fc5668e5a521e5b45e8370a62">mlx::core::astype</a> (<a class="el" href="classmlx_1_1core_1_1array.html">array</a> a, <a class="el" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:ga0e58c24fc5668e5a521e5b45e8370a62"><td class="mdescLeft">&#160;</td><td class="mdescRight">Convert an array to the given data type. <br /></td></tr>
<tr class="separator:ga0e58c24fc5668e5a521e5b45e8370a62"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga8de80ecef30fc560003d40f61a38b99d" id="r_ga8de80ecef30fc560003d40f61a38b99d"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga8de80ecef30fc560003d40f61a38b99d">mlx::core::as_strided</a> (<a class="el" href="classmlx_1_1core_1_1array.html">array</a> a, std::vector&lt; int &gt; shape, std::vector&lt; size_t &gt; strides, size_t offset, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:ga8de80ecef30fc560003d40f61a38b99d"><td class="mdescLeft">&#160;</td><td class="mdescRight">Create a view of an array with the given shape and strides. <br /></td></tr>
<tr class="separator:ga8de80ecef30fc560003d40f61a38b99d"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga6085b03f2662ef2a61de523fd609f3bf" id="r_ga6085b03f2662ef2a61de523fd609f3bf"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga6085b03f2662ef2a61de523fd609f3bf">mlx::core::as_strided</a> (<a class="el" href="classmlx_1_1core_1_1array.html">array</a> a, <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> shape, <a class="el" href="namespacemlx_1_1core.html#aca6e505f52ea1464d9c2c09f9f445d62">Strides</a> strides, size_t offset, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:ga6085b03f2662ef2a61de523fd609f3bf"><td class="mdescLeft">&#160;</td><td class="mdescRight">Create a view of an array with the given shape and strides. <br /></td></tr>
<tr class="separator:ga6085b03f2662ef2a61de523fd609f3bf"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:gae306e93af12f774bd80bad6c231b09d6" id="r_gae306e93af12f774bd80bad6c231b09d6"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#gae306e93af12f774bd80bad6c231b09d6">mlx::core::copy</a> (<a class="el" href="classmlx_1_1core_1_1array.html">array</a> a, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:gae306e93af12f774bd80bad6c231b09d6"><td class="mdescLeft">&#160;</td><td class="mdescRight"><a class="el" href="classmlx_1_1core_1_1_copy.html">Copy</a> another array. <br /></td></tr>
<tr class="separator:gae306e93af12f774bd80bad6c231b09d6"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga7e85a476530c96ab73b894d1bddacf84" id="r_ga7e85a476530c96ab73b894d1bddacf84"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga7e85a476530c96ab73b894d1bddacf84">mlx::core::full</a> (std::vector&lt; int &gt; shape, <a class="el" href="classmlx_1_1core_1_1array.html">array</a> vals, <a class="el" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:ga7e85a476530c96ab73b894d1bddacf84"><td class="mdescLeft">&#160;</td><td class="mdescRight">Fill an array of the given shape with the given value(s). <br /></td></tr>
<tr class="separator:ga7e85a476530c96ab73b894d1bddacf84"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga9e3b15c3ec6b7922296777c5e1878d52" id="r_ga9e3b15c3ec6b7922296777c5e1878d52"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga9e3b15c3ec6b7922296777c5e1878d52">mlx::core::full</a> (std::vector&lt; int &gt; shape, <a class="el" href="classmlx_1_1core_1_1array.html">array</a> vals, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:ga9e3b15c3ec6b7922296777c5e1878d52"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:gac34ff7f1e217b93e878053aaa5a51e58" id="r_gac34ff7f1e217b93e878053aaa5a51e58"><td class="memTemplParams" colspan="2">template&lt;typename T &gt; </td></tr>
<tr class="memitem:gac34ff7f1e217b93e878053aaa5a51e58"><td class="memTemplItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="group__ops.html#gac34ff7f1e217b93e878053aaa5a51e58">mlx::core::full</a> (std::vector&lt; int &gt; shape, T val, <a class="el" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:gac34ff7f1e217b93e878053aaa5a51e58"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga5e33cf83db1a65c35e688142bd41d35c" id="r_ga5e33cf83db1a65c35e688142bd41d35c"><td class="memTemplParams" colspan="2">template&lt;typename T &gt; </td></tr>
<tr class="memitem:ga5e33cf83db1a65c35e688142bd41d35c"><td class="memTemplItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="group__ops.html#ga5e33cf83db1a65c35e688142bd41d35c">mlx::core::full</a> (std::vector&lt; int &gt; shape, T val, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:ga5e33cf83db1a65c35e688142bd41d35c"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga34573eb2ead179839579142f3aedb3b8" id="r_ga34573eb2ead179839579142f3aedb3b8"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga34573eb2ead179839579142f3aedb3b8">mlx::core::zeros</a> (const std::vector&lt; int &gt; &amp;shape, <a class="el" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:ga34573eb2ead179839579142f3aedb3b8"><td class="mdescLeft">&#160;</td><td class="mdescRight">Fill an array of the given shape with zeros. <br /></td></tr>
<tr class="separator:ga34573eb2ead179839579142f3aedb3b8"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga7c1487f8a830c9df75a961b3a24527ee" id="r_ga7c1487f8a830c9df75a961b3a24527ee"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga7c1487f8a830c9df75a961b3a24527ee">mlx::core::zeros</a> (const std::vector&lt; int &gt; &amp;shape, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:ga7c1487f8a830c9df75a961b3a24527ee"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga1cf232308668fe3f4214c8b895ed4aee" id="r_ga1cf232308668fe3f4214c8b895ed4aee"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga1cf232308668fe3f4214c8b895ed4aee">mlx::core::full</a> (<a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> shape, <a class="el" href="classmlx_1_1core_1_1array.html">array</a> vals, <a class="el" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:ga1cf232308668fe3f4214c8b895ed4aee"><td class="mdescLeft">&#160;</td><td class="mdescRight">Fill an array of the given shape with the given value(s). <br /></td></tr>
<tr class="separator:ga1cf232308668fe3f4214c8b895ed4aee"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga59f6c844cbb173e108c3eeb11801f8c6" id="r_ga59f6c844cbb173e108c3eeb11801f8c6"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga59f6c844cbb173e108c3eeb11801f8c6">mlx::core::full</a> (<a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> shape, <a class="el" href="classmlx_1_1core_1_1array.html">array</a> vals, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:ga59f6c844cbb173e108c3eeb11801f8c6"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:gaf073760b7b51fe35932da0d81c531a55" id="r_gaf073760b7b51fe35932da0d81c531a55"><td class="memTemplParams" colspan="2">template&lt;typename T &gt; </td></tr>
<tr class="memitem:gaf073760b7b51fe35932da0d81c531a55"><td class="memTemplItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="group__ops.html#gaf073760b7b51fe35932da0d81c531a55">mlx::core::full</a> (<a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> shape, T val, <a class="el" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:gaf073760b7b51fe35932da0d81c531a55"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:gaf6f2cce92aff9b71756a3cc3c961fd5a" id="r_gaf6f2cce92aff9b71756a3cc3c961fd5a"><td class="memTemplParams" colspan="2">template&lt;typename T &gt; </td></tr>
<tr class="memitem:gaf6f2cce92aff9b71756a3cc3c961fd5a"><td class="memTemplItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="group__ops.html#gaf6f2cce92aff9b71756a3cc3c961fd5a">mlx::core::full</a> (<a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> shape, T val, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:gaf6f2cce92aff9b71756a3cc3c961fd5a"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:gae2cace3b388cec4e520659a91879e1c1" id="r_gae2cace3b388cec4e520659a91879e1c1"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#gae2cace3b388cec4e520659a91879e1c1">mlx::core::zeros</a> (const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &amp;shape, <a class="el" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:gae2cace3b388cec4e520659a91879e1c1"><td class="mdescLeft">&#160;</td><td class="mdescRight">Fill an array of the given shape with zeros. <br /></td></tr>
<tr class="separator:gae2cace3b388cec4e520659a91879e1c1"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:gac8aa722f5e798819b7091693173f1f36" id="r_gac8aa722f5e798819b7091693173f1f36"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#gac8aa722f5e798819b7091693173f1f36">mlx::core::zeros</a> (const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &amp;shape, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:gac8aa722f5e798819b7091693173f1f36"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:gafbb857094d784b38c78683a091ffdbde" id="r_gafbb857094d784b38c78683a091ffdbde"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#gafbb857094d784b38c78683a091ffdbde">mlx::core::zeros_like</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:gafbb857094d784b38c78683a091ffdbde"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga54eeed455321a54c8e72e16552a978f2" id="r_ga54eeed455321a54c8e72e16552a978f2"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga54eeed455321a54c8e72e16552a978f2">mlx::core::ones</a> (const std::vector&lt; int &gt; &amp;shape, <a class="el" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:ga54eeed455321a54c8e72e16552a978f2"><td class="mdescLeft">&#160;</td><td class="mdescRight">Fill an array of the given shape with ones. <br /></td></tr>
<tr class="separator:ga54eeed455321a54c8e72e16552a978f2"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga6cf4b5e8580e4436302c519d05897dab" id="r_ga6cf4b5e8580e4436302c519d05897dab"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga6cf4b5e8580e4436302c519d05897dab">mlx::core::ones</a> (const std::vector&lt; int &gt; &amp;shape, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:ga6cf4b5e8580e4436302c519d05897dab"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:gae0069146cf8c819b15ba29aa7231a3f0" id="r_gae0069146cf8c819b15ba29aa7231a3f0"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#gae0069146cf8c819b15ba29aa7231a3f0">mlx::core::ones</a> (const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &amp;shape, <a class="el" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:gae0069146cf8c819b15ba29aa7231a3f0"><td class="mdescLeft">&#160;</td><td class="mdescRight">Fill an array of the given shape with ones. <br /></td></tr>
<tr class="separator:gae0069146cf8c819b15ba29aa7231a3f0"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:gace4cf016562af58991f9f961170e156f" id="r_gace4cf016562af58991f9f961170e156f"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#gace4cf016562af58991f9f961170e156f">mlx::core::ones</a> (const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &amp;shape, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:gace4cf016562af58991f9f961170e156f"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga94f8d3b1906fee99da9cbe39f7be7d42" id="r_ga94f8d3b1906fee99da9cbe39f7be7d42"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga94f8d3b1906fee99da9cbe39f7be7d42">mlx::core::ones_like</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:ga94f8d3b1906fee99da9cbe39f7be7d42"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga45e9e68246b0d1cf03c3cc9c9e7e6ae3" id="r_ga45e9e68246b0d1cf03c3cc9c9e7e6ae3"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3">mlx::core::eye</a> (int n, int m, int k, <a class="el" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
@@ -191,9 +191,9 @@ Functions</h2></td></tr>
<tr class="separator:ga83e0bb45dc770cf014531d873b78c5a2"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:gaa9df5917876eeb0cb28b7fa81f880412" id="r_gaa9df5917876eeb0cb28b7fa81f880412"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#gaa9df5917876eeb0cb28b7fa81f880412">mlx::core::triu</a> (<a class="el" href="classmlx_1_1core_1_1array.html">array</a> x, int k=0, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:gaa9df5917876eeb0cb28b7fa81f880412"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga7b0e3d46a39d99b1628b422049790082" id="r_ga7b0e3d46a39d99b1628b422049790082"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga7b0e3d46a39d99b1628b422049790082">mlx::core::reshape</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, std::vector&lt; int &gt; shape, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:ga7b0e3d46a39d99b1628b422049790082"><td class="mdescLeft">&#160;</td><td class="mdescRight"><a class="el" href="classmlx_1_1core_1_1_reshape.html">Reshape</a> an array to the given shape. <br /></td></tr>
<tr class="separator:ga7b0e3d46a39d99b1628b422049790082"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga084f03ce2b22258afb7c8b45e17af828" id="r_ga084f03ce2b22258afb7c8b45e17af828"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga084f03ce2b22258afb7c8b45e17af828">mlx::core::reshape</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> shape, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:ga084f03ce2b22258afb7c8b45e17af828"><td class="mdescLeft">&#160;</td><td class="mdescRight"><a class="el" href="classmlx_1_1core_1_1_reshape.html">Reshape</a> an array to the given shape. <br /></td></tr>
<tr class="separator:ga084f03ce2b22258afb7c8b45e17af828"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga50aa98754b412bb57c083f6e3e95061f" id="r_ga50aa98754b412bb57c083f6e3e95061f"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga50aa98754b412bb57c083f6e3e95061f">mlx::core::flatten</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, int start_axis, int end_axis=-1, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:ga50aa98754b412bb57c083f6e3e95061f"><td class="mdescLeft">&#160;</td><td class="mdescRight">Flatten the dimensions in the range <code>[start_axis, end_axis]</code> . <br /></td></tr>
<tr class="separator:ga50aa98754b412bb57c083f6e3e95061f"><td class="memSeparator" colspan="2">&#160;</td></tr>
@@ -218,18 +218,18 @@ Functions</h2></td></tr>
<tr class="memitem:ga7a80adb4a5a36d18b5f234d4b034950a" id="r_ga7a80adb4a5a36d18b5f234d4b034950a"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga7a80adb4a5a36d18b5f234d4b034950a">mlx::core::expand_dims</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, int axis, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:ga7a80adb4a5a36d18b5f234d4b034950a"><td class="mdescLeft">&#160;</td><td class="mdescRight"><a class="el" href="classmlx_1_1core_1_1_add.html">Add</a> a singleton dimension at the given axis. <br /></td></tr>
<tr class="separator:ga7a80adb4a5a36d18b5f234d4b034950a"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:gad66135407dbb41b3c5d2cdfd51226c21" id="r_gad66135407dbb41b3c5d2cdfd51226c21"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#gad66135407dbb41b3c5d2cdfd51226c21">mlx::core::slice</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, std::vector&lt; int &gt; start, std::vector&lt; int &gt; stop, std::vector&lt; int &gt; strides, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:gad66135407dbb41b3c5d2cdfd51226c21"><td class="mdescLeft">&#160;</td><td class="mdescRight"><a class="el" href="classmlx_1_1core_1_1_slice.html">Slice</a> an array. <br /></td></tr>
<tr class="separator:gad66135407dbb41b3c5d2cdfd51226c21"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:gaa97ce866c5e38b92b093e9321affcc57" id="r_gaa97ce866c5e38b92b093e9321affcc57"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#gaa97ce866c5e38b92b093e9321affcc57">mlx::core::slice</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, std::vector&lt; int &gt; start, std::vector&lt; int &gt; stop, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:gaa97ce866c5e38b92b093e9321affcc57"><td class="mdescLeft">&#160;</td><td class="mdescRight"><a class="el" href="classmlx_1_1core_1_1_slice.html">Slice</a> an array with a stride of 1 in each dimension. <br /></td></tr>
<tr class="separator:gaa97ce866c5e38b92b093e9321affcc57"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga3875660e4ce2c8add8bfcf8144078708" id="r_ga3875660e4ce2c8add8bfcf8144078708"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga3875660e4ce2c8add8bfcf8144078708">mlx::core::slice_update</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;src, const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;update, std::vector&lt; int &gt; start, std::vector&lt; int &gt; stop, std::vector&lt; int &gt; strides, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:ga3875660e4ce2c8add8bfcf8144078708"><td class="mdescLeft">&#160;</td><td class="mdescRight">Update a slice from the source array. <br /></td></tr>
<tr class="separator:ga3875660e4ce2c8add8bfcf8144078708"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga03ffbbb4d989a463ef43f41ebf7eabef" id="r_ga03ffbbb4d989a463ef43f41ebf7eabef"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga03ffbbb4d989a463ef43f41ebf7eabef">mlx::core::slice_update</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;src, const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;update, std::vector&lt; int &gt; start, std::vector&lt; int &gt; stop, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:ga03ffbbb4d989a463ef43f41ebf7eabef"><td class="mdescLeft">&#160;</td><td class="mdescRight">Update a slice from the source array with stride 1 in each dimension. <br /></td></tr>
<tr class="separator:ga03ffbbb4d989a463ef43f41ebf7eabef"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga29718cd5005dbcde0396b6fd65cc041d" id="r_ga29718cd5005dbcde0396b6fd65cc041d"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga29718cd5005dbcde0396b6fd65cc041d">mlx::core::slice</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> start, <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> stop, <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> strides, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:ga29718cd5005dbcde0396b6fd65cc041d"><td class="mdescLeft">&#160;</td><td class="mdescRight"><a class="el" href="classmlx_1_1core_1_1_slice.html">Slice</a> an array. <br /></td></tr>
<tr class="separator:ga29718cd5005dbcde0396b6fd65cc041d"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:gaec56dcb94d5e7f7b885fb60b4bf4aa9d" id="r_gaec56dcb94d5e7f7b885fb60b4bf4aa9d"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#gaec56dcb94d5e7f7b885fb60b4bf4aa9d">mlx::core::slice</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> start, <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> stop, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:gaec56dcb94d5e7f7b885fb60b4bf4aa9d"><td class="mdescLeft">&#160;</td><td class="mdescRight"><a class="el" href="classmlx_1_1core_1_1_slice.html">Slice</a> an array with a stride of 1 in each dimension. <br /></td></tr>
<tr class="separator:gaec56dcb94d5e7f7b885fb60b4bf4aa9d"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga7b7e786985d27789aaed20bb2f9509be" id="r_ga7b7e786985d27789aaed20bb2f9509be"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga7b7e786985d27789aaed20bb2f9509be">mlx::core::slice_update</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;src, const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;update, <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> start, <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> stop, <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> strides, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:ga7b7e786985d27789aaed20bb2f9509be"><td class="mdescLeft">&#160;</td><td class="mdescRight">Update a slice from the source array. <br /></td></tr>
<tr class="separator:ga7b7e786985d27789aaed20bb2f9509be"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga3b7c6136ae5a38dd6457b65c1833aa67" id="r_ga3b7c6136ae5a38dd6457b65c1833aa67"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga3b7c6136ae5a38dd6457b65c1833aa67">mlx::core::slice_update</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;src, const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;update, <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> start, <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> stop, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:ga3b7c6136ae5a38dd6457b65c1833aa67"><td class="mdescLeft">&#160;</td><td class="mdescRight">Update a slice from the source array with stride 1 in each dimension. <br /></td></tr>
<tr class="separator:ga3b7c6136ae5a38dd6457b65c1833aa67"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga7534290bceab5fb3831a05d67bebce7d" id="r_ga7534290bceab5fb3831a05d67bebce7d"><td class="memItemLeft" align="right" valign="top">std::vector&lt; <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &gt;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga7534290bceab5fb3831a05d67bebce7d">mlx::core::split</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, int num_splits, int axis, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:ga7534290bceab5fb3831a05d67bebce7d"><td class="mdescLeft">&#160;</td><td class="mdescRight"><a class="el" href="classmlx_1_1core_1_1_split.html">Split</a> an array into sub-arrays along a given axis. <br /></td></tr>
<tr class="separator:ga7534290bceab5fb3831a05d67bebce7d"><td class="memSeparator" colspan="2">&#160;</td></tr>
@@ -286,9 +286,9 @@ Functions</h2></td></tr>
<tr class="memitem:ga68da0176fefbe0c0096783c6fd926c6a" id="r_ga68da0176fefbe0c0096783c6fd926c6a"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga68da0176fefbe0c0096783c6fd926c6a">mlx::core::transpose</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:ga68da0176fefbe0c0096783c6fd926c6a"><td class="mdescLeft">&#160;</td><td class="mdescRight">Permutes the dimensions in reverse order. <br /></td></tr>
<tr class="separator:ga68da0176fefbe0c0096783c6fd926c6a"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:gad256e86cc1a6e6b3832e392baa90318d" id="r_gad256e86cc1a6e6b3832e392baa90318d"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#gad256e86cc1a6e6b3832e392baa90318d">mlx::core::broadcast_to</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, const std::vector&lt; int &gt; &amp;shape, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:gad256e86cc1a6e6b3832e392baa90318d"><td class="mdescLeft">&#160;</td><td class="mdescRight"><a class="el" href="classmlx_1_1core_1_1_broadcast.html">Broadcast</a> an array to a given shape. <br /></td></tr>
<tr class="separator:gad256e86cc1a6e6b3832e392baa90318d"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga2fd5891f11593b7f09550e884f969013" id="r_ga2fd5891f11593b7f09550e884f969013"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga2fd5891f11593b7f09550e884f969013">mlx::core::broadcast_to</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &amp;shape, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:ga2fd5891f11593b7f09550e884f969013"><td class="mdescLeft">&#160;</td><td class="mdescRight"><a class="el" href="classmlx_1_1core_1_1_broadcast.html">Broadcast</a> an array to a given shape. <br /></td></tr>
<tr class="separator:ga2fd5891f11593b7f09550e884f969013"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:gab783890428b596f715dc7dd2057eae99" id="r_gab783890428b596f715dc7dd2057eae99"><td class="memItemLeft" align="right" valign="top">std::vector&lt; <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &gt;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#gab783890428b596f715dc7dd2057eae99">mlx::core::broadcast_arrays</a> (const std::vector&lt; <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &gt; &amp;inputs, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:gab783890428b596f715dc7dd2057eae99"><td class="mdescLeft">&#160;</td><td class="mdescRight"><a class="el" href="classmlx_1_1core_1_1_broadcast.html">Broadcast</a> a vector of arrays against one another. <br /></td></tr>
<tr class="separator:gab783890428b596f715dc7dd2057eae99"><td class="memSeparator" colspan="2">&#160;</td></tr>
@@ -728,11 +728,11 @@ Functions</h2></td></tr>
<tr class="memitem:ga753d59f5a9f5f2362865ee83b4dced2a" id="r_ga753d59f5a9f5f2362865ee83b4dced2a"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga753d59f5a9f5f2362865ee83b4dced2a">mlx::core::matmul</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;b, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:ga753d59f5a9f5f2362865ee83b4dced2a"><td class="mdescLeft">&#160;</td><td class="mdescRight">Matrix-matrix multiplication. <br /></td></tr>
<tr class="separator:ga753d59f5a9f5f2362865ee83b4dced2a"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:gab6e7f655a9ff15350ca5379692f9d444" id="r_gab6e7f655a9ff15350ca5379692f9d444"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#gab6e7f655a9ff15350ca5379692f9d444">mlx::core::gather</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, const std::vector&lt; <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &gt; &amp;indices, const std::vector&lt; int &gt; &amp;axes, const std::vector&lt; int &gt; &amp;slice_sizes, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:gab6e7f655a9ff15350ca5379692f9d444"><td class="mdescLeft">&#160;</td><td class="mdescRight"><a class="el" href="classmlx_1_1core_1_1_gather.html">Gather</a> array entries given indices and slices. <br /></td></tr>
<tr class="separator:gab6e7f655a9ff15350ca5379692f9d444"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:gadb4337ca5d4f88fe9e7c083bc478158b" id="r_gadb4337ca5d4f88fe9e7c083bc478158b"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#gadb4337ca5d4f88fe9e7c083bc478158b">mlx::core::gather</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;indices, int axis, const std::vector&lt; int &gt; &amp;slice_sizes, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:gadb4337ca5d4f88fe9e7c083bc478158b"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga8fcc3ad0677c834c36b72d5b2ebba6d0" id="r_ga8fcc3ad0677c834c36b72d5b2ebba6d0"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga8fcc3ad0677c834c36b72d5b2ebba6d0">mlx::core::gather</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, const std::vector&lt; <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &gt; &amp;indices, const std::vector&lt; int &gt; &amp;axes, const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &amp;slice_sizes, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:ga8fcc3ad0677c834c36b72d5b2ebba6d0"><td class="mdescLeft">&#160;</td><td class="mdescRight"><a class="el" href="classmlx_1_1core_1_1_gather.html">Gather</a> array entries given indices and slices. <br /></td></tr>
<tr class="separator:ga8fcc3ad0677c834c36b72d5b2ebba6d0"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:gafe2bd174c9953ed7f12664f7abaca0e6" id="r_gafe2bd174c9953ed7f12664f7abaca0e6"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#gafe2bd174c9953ed7f12664f7abaca0e6">mlx::core::gather</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;indices, int axis, const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &amp;slice_sizes, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:gafe2bd174c9953ed7f12664f7abaca0e6"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:gac2fc270882fcfa81eb8bd068cc0d86d7" id="r_gac2fc270882fcfa81eb8bd068cc0d86d7"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#gac2fc270882fcfa81eb8bd068cc0d86d7">mlx::core::take</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;indices, int axis, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:gac2fc270882fcfa81eb8bd068cc0d86d7"><td class="mdescLeft">&#160;</td><td class="mdescRight">Take array slices at the given indices of the specified axis. <br /></td></tr>
<tr class="separator:gac2fc270882fcfa81eb8bd068cc0d86d7"><td class="memSeparator" colspan="2">&#160;</td></tr>
@@ -924,16 +924,16 @@ Functions</h2></td></tr>
<tr class="memitem:gac40e48c69f9c715a767912c30836e75c" id="r_gac40e48c69f9c715a767912c30836e75c"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c">mlx::core::roll</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, int shift, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="memdesc:gac40e48c69f9c715a767912c30836e75c"><td class="mdescLeft">&#160;</td><td class="mdescRight">Roll elements along an axis and introduce them on the other side. <br /></td></tr>
<tr class="separator:gac40e48c69f9c715a767912c30836e75c"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:gab30fb942bb6ee00bb50da04033f949a9" id="r_gab30fb942bb6ee00bb50da04033f949a9"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#gab30fb942bb6ee00bb50da04033f949a9">mlx::core::roll</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, const std::vector&lt; int &gt; &amp;shift, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:gab30fb942bb6ee00bb50da04033f949a9"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga5011d1a5735c64e5b91afa56c7e2cc02" id="r_ga5011d1a5735c64e5b91afa56c7e2cc02"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga5011d1a5735c64e5b91afa56c7e2cc02">mlx::core::roll</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &amp;shift, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:ga5011d1a5735c64e5b91afa56c7e2cc02"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga8694ec137165752cb6d8a36a6b7c3436" id="r_ga8694ec137165752cb6d8a36a6b7c3436"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga8694ec137165752cb6d8a36a6b7c3436">mlx::core::roll</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, int shift, int axis, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:ga8694ec137165752cb6d8a36a6b7c3436"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga665f502ecc96f1f4467556b784abf9ae" id="r_ga665f502ecc96f1f4467556b784abf9ae"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga665f502ecc96f1f4467556b784abf9ae">mlx::core::roll</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, int shift, const std::vector&lt; int &gt; &amp;axes, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:ga665f502ecc96f1f4467556b784abf9ae"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga4c335f86ff2a05daf8dd8aedc628aa59" id="r_ga4c335f86ff2a05daf8dd8aedc628aa59"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga4c335f86ff2a05daf8dd8aedc628aa59">mlx::core::roll</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, const std::vector&lt; int &gt; &amp;shift, int axis, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:ga4c335f86ff2a05daf8dd8aedc628aa59"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga6c6fe942a8f7c573a1cb2342566f74d2" id="r_ga6c6fe942a8f7c573a1cb2342566f74d2"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga6c6fe942a8f7c573a1cb2342566f74d2">mlx::core::roll</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, const std::vector&lt; int &gt; &amp;shift, const std::vector&lt; int &gt; &amp;axes, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:ga6c6fe942a8f7c573a1cb2342566f74d2"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga5e9833b1b7a1c636db6f7ad9809ce902" id="r_ga5e9833b1b7a1c636db6f7ad9809ce902"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga5e9833b1b7a1c636db6f7ad9809ce902">mlx::core::roll</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, int shift, const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &amp;axes, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:ga5e9833b1b7a1c636db6f7ad9809ce902"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga79137f90bc44ac9e35f408c012701df9" id="r_ga79137f90bc44ac9e35f408c012701df9"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga79137f90bc44ac9e35f408c012701df9">mlx::core::roll</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &amp;shift, int axis, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:ga79137f90bc44ac9e35f408c012701df9"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga9d76930fb567a7d459ff96fb851abe36" id="r_ga9d76930fb567a7d459ff96fb851abe36"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga9d76930fb567a7d459ff96fb851abe36">mlx::core::roll</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &amp;shift, const std::vector&lt; int &gt; &amp;axes, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:ga9d76930fb567a7d459ff96fb851abe36"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:gaf8913cabeb9fb193ba687aaeb2087764" id="r_gaf8913cabeb9fb193ba687aaeb2087764"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#gaf8913cabeb9fb193ba687aaeb2087764">mlx::core::real</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:gaf8913cabeb9fb193ba687aaeb2087764"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga7ff592a64d528f0cf4f3d098465da029" id="r_ga7ff592a64d528f0cf4f3d098465da029"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__ops.html#ga7ff592a64d528f0cf4f3d098465da029">mlx::core::imag</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;a, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>