mirror of
https://github.com/ml-explore/mlx.git
synced 2025-09-18 10:26:56 +08:00
rebase
This commit is contained in:
268
docs/build/html/group__ops.html
vendored
268
docs/build/html/group__ops.html
vendored
@@ -117,35 +117,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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#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"> </td><td class="mdescRight">Convert an array to the given data type. <br /></td></tr>
|
||||
<tr class="separator:ga0e58c24fc5668e5a521e5b45e8370a62"><td class="memSeparator" colspan="2"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga8de80ecef30fc560003d40f61a38b99d">mlx::core::as_strided</a> (<a class="el" href="classmlx_1_1core_1_1array.html">array</a> a, std::vector< int > shape, std::vector< size_t > 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"> </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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#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"> </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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#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"> </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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga7e85a476530c96ab73b894d1bddacf84">mlx::core::full</a> (std::vector< int > 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"> </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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga9e3b15c3ec6b7922296777c5e1878d52">mlx::core::full</a> (std::vector< int > 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"> </td></tr>
|
||||
<tr class="memitem:gac34ff7f1e217b93e878053aaa5a51e58" id="r_gac34ff7f1e217b93e878053aaa5a51e58"><td class="memTemplParams" colspan="2">template<typename T > </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> </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#gac34ff7f1e217b93e878053aaa5a51e58">mlx::core::full</a> (std::vector< int > 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"> </td></tr>
|
||||
<tr class="memitem:ga5e33cf83db1a65c35e688142bd41d35c" id="r_ga5e33cf83db1a65c35e688142bd41d35c"><td class="memTemplParams" colspan="2">template<typename T > </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> </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#ga5e33cf83db1a65c35e688142bd41d35c">mlx::core::full</a> (std::vector< int > 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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga34573eb2ead179839579142f3aedb3b8">mlx::core::zeros</a> (const std::vector< int > &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"> </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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga7c1487f8a830c9df75a961b3a24527ee">mlx::core::zeros</a> (const std::vector< int > &shape, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="separator:ga7c1487f8a830c9df75a961b3a24527ee"><td class="memSeparator" colspan="2"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#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"> </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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#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"> </td></tr>
|
||||
<tr class="memitem:gaf073760b7b51fe35932da0d81c531a55" id="r_gaf073760b7b51fe35932da0d81c531a55"><td class="memTemplParams" colspan="2">template<typename T > </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> </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#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"> </td></tr>
|
||||
<tr class="memitem:gaf6f2cce92aff9b71756a3cc3c961fd5a" id="r_gaf6f2cce92aff9b71756a3cc3c961fd5a"><td class="memTemplParams" colspan="2">template<typename T > </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> </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#gae2cace3b388cec4e520659a91879e1c1">mlx::core::zeros</a> (const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &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"> </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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#gac8aa722f5e798819b7091693173f1f36">mlx::core::zeros</a> (const <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="separator:gac8aa722f5e798819b7091693173f1f36"><td class="memSeparator" colspan="2"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#gafbb857094d784b38c78683a091ffdbde">mlx::core::zeros_like</a> (const <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="separator:gafbb857094d784b38c78683a091ffdbde"><td class="memSeparator" colspan="2"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga54eeed455321a54c8e72e16552a978f2">mlx::core::ones</a> (const std::vector< int > &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"> </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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga6cf4b5e8580e4436302c519d05897dab">mlx::core::ones</a> (const std::vector< int > &shape, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="separator:ga6cf4b5e8580e4436302c519d05897dab"><td class="memSeparator" colspan="2"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#gae0069146cf8c819b15ba29aa7231a3f0">mlx::core::ones</a> (const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &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"> </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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#gace4cf016562af58991f9f961170e156f">mlx::core::ones</a> (const <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="separator:gace4cf016562af58991f9f961170e156f"><td class="memSeparator" colspan="2"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga94f8d3b1906fee99da9cbe39f7be7d42">mlx::core::ones_like</a> (const <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="separator:ga94f8d3b1906fee99da9cbe39f7be7d42"><td class="memSeparator" colspan="2"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#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>
|
||||
@@ -172,9 +172,9 @@ Functions</h2></td></tr>
|
||||
<tr class="separator:ga83e0bb45dc770cf014531d873b78c5a2"><td class="memSeparator" colspan="2"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga7b0e3d46a39d99b1628b422049790082">mlx::core::reshape</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &a, std::vector< int > shape, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="memdesc:ga7b0e3d46a39d99b1628b422049790082"><td class="mdescLeft"> </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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga084f03ce2b22258afb7c8b45e17af828">mlx::core::reshape</a> (const <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#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="memdesc:ga084f03ce2b22258afb7c8b45e17af828"><td class="mdescLeft"> </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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga50aa98754b412bb57c083f6e3e95061f">mlx::core::flatten</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &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"> </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"> </td></tr>
|
||||
@@ -199,18 +199,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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga7a80adb4a5a36d18b5f234d4b034950a">mlx::core::expand_dims</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &a, int axis, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="memdesc:ga7a80adb4a5a36d18b5f234d4b034950a"><td class="mdescLeft"> </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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#gad66135407dbb41b3c5d2cdfd51226c21">mlx::core::slice</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &a, std::vector< int > start, std::vector< int > stop, std::vector< int > strides, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="memdesc:gad66135407dbb41b3c5d2cdfd51226c21"><td class="mdescLeft"> </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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#gaa97ce866c5e38b92b093e9321affcc57">mlx::core::slice</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &a, std::vector< int > start, std::vector< int > stop, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="memdesc:gaa97ce866c5e38b92b093e9321affcc57"><td class="mdescLeft"> </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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga3875660e4ce2c8add8bfcf8144078708">mlx::core::slice_update</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &src, const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &update, std::vector< int > start, std::vector< int > stop, std::vector< int > strides, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="memdesc:ga3875660e4ce2c8add8bfcf8144078708"><td class="mdescLeft"> </td><td class="mdescRight">Update a slice from the source array. <br /></td></tr>
|
||||
<tr class="separator:ga3875660e4ce2c8add8bfcf8144078708"><td class="memSeparator" colspan="2"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga03ffbbb4d989a463ef43f41ebf7eabef">mlx::core::slice_update</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &src, const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &update, std::vector< int > start, std::vector< int > stop, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="memdesc:ga03ffbbb4d989a463ef43f41ebf7eabef"><td class="mdescLeft"> </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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga29718cd5005dbcde0396b6fd65cc041d">mlx::core::slice</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &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"> </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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#gaec56dcb94d5e7f7b885fb60b4bf4aa9d">mlx::core::slice</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &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"> </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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga7b7e786985d27789aaed20bb2f9509be">mlx::core::slice_update</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &src, const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &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"> </td><td class="mdescRight">Update a slice from the source array. <br /></td></tr>
|
||||
<tr class="separator:ga7b7e786985d27789aaed20bb2f9509be"><td class="memSeparator" colspan="2"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga3b7c6136ae5a38dd6457b65c1833aa67">mlx::core::slice_update</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &src, const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &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"> </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"> </td></tr>
|
||||
<tr class="memitem:ga7534290bceab5fb3831a05d67bebce7d" id="r_ga7534290bceab5fb3831a05d67bebce7d"><td class="memItemLeft" align="right" valign="top">std::vector< <a class="el" href="classmlx_1_1core_1_1array.html">array</a> > </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga7534290bceab5fb3831a05d67bebce7d">mlx::core::split</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &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"> </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"> </td></tr>
|
||||
@@ -267,9 +267,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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga68da0176fefbe0c0096783c6fd926c6a">mlx::core::transpose</a> (const <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:ga68da0176fefbe0c0096783c6fd926c6a"><td class="mdescLeft"> </td><td class="mdescRight">Permutes the dimensions in reverse order. <br /></td></tr>
|
||||
<tr class="separator:ga68da0176fefbe0c0096783c6fd926c6a"><td class="memSeparator" colspan="2"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#gad256e86cc1a6e6b3832e392baa90318d">mlx::core::broadcast_to</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &a, const std::vector< int > &shape, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="memdesc:gad256e86cc1a6e6b3832e392baa90318d"><td class="mdescLeft"> </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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga2fd5891f11593b7f09550e884f969013">mlx::core::broadcast_to</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &a, const <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:ga2fd5891f11593b7f09550e884f969013"><td class="mdescLeft"> </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"> </td></tr>
|
||||
<tr class="memitem:gab783890428b596f715dc7dd2057eae99" id="r_gab783890428b596f715dc7dd2057eae99"><td class="memItemLeft" align="right" valign="top">std::vector< <a class="el" href="classmlx_1_1core_1_1array.html">array</a> > </td><td class="memItemRight" valign="bottom"><a class="el" href="#gab783890428b596f715dc7dd2057eae99">mlx::core::broadcast_arrays</a> (const std::vector< <a class="el" href="classmlx_1_1core_1_1array.html">array</a> > &inputs, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="memdesc:gab783890428b596f715dc7dd2057eae99"><td class="mdescLeft"> </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"> </td></tr>
|
||||
@@ -709,11 +709,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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga753d59f5a9f5f2362865ee83b4dced2a">mlx::core::matmul</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &a, const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &b, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="memdesc:ga753d59f5a9f5f2362865ee83b4dced2a"><td class="mdescLeft"> </td><td class="mdescRight">Matrix-matrix multiplication. <br /></td></tr>
|
||||
<tr class="separator:ga753d59f5a9f5f2362865ee83b4dced2a"><td class="memSeparator" colspan="2"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#gab6e7f655a9ff15350ca5379692f9d444">mlx::core::gather</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &a, const std::vector< <a class="el" href="classmlx_1_1core_1_1array.html">array</a> > &indices, const std::vector< int > &axes, const std::vector< int > &slice_sizes, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="memdesc:gab6e7f655a9ff15350ca5379692f9d444"><td class="mdescLeft"> </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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#gadb4337ca5d4f88fe9e7c083bc478158b">mlx::core::gather</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &a, const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &indices, int axis, const std::vector< int > &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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga8fcc3ad0677c834c36b72d5b2ebba6d0">mlx::core::gather</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &a, const std::vector< <a class="el" href="classmlx_1_1core_1_1array.html">array</a> > &indices, const std::vector< int > &axes, const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &slice_sizes, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="memdesc:ga8fcc3ad0677c834c36b72d5b2ebba6d0"><td class="mdescLeft"> </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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#gafe2bd174c9953ed7f12664f7abaca0e6">mlx::core::gather</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &a, const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &indices, int axis, const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#gac2fc270882fcfa81eb8bd068cc0d86d7">mlx::core::take</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &a, const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &indices, int axis, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="memdesc:gac2fc270882fcfa81eb8bd068cc0d86d7"><td class="mdescLeft"> </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"> </td></tr>
|
||||
@@ -905,16 +905,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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#gac40e48c69f9c715a767912c30836e75c">mlx::core::roll</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &a, int shift, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="memdesc:gac40e48c69f9c715a767912c30836e75c"><td class="mdescLeft"> </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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#gab30fb942bb6ee00bb50da04033f949a9">mlx::core::roll</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &a, const std::vector< int > &shift, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="separator:gab30fb942bb6ee00bb50da04033f949a9"><td class="memSeparator" colspan="2"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga5011d1a5735c64e5b91afa56c7e2cc02">mlx::core::roll</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &a, const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &shift, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="separator:ga5011d1a5735c64e5b91afa56c7e2cc02"><td class="memSeparator" colspan="2"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga8694ec137165752cb6d8a36a6b7c3436">mlx::core::roll</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga665f502ecc96f1f4467556b784abf9ae">mlx::core::roll</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &a, int shift, const std::vector< int > &axes, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="separator:ga665f502ecc96f1f4467556b784abf9ae"><td class="memSeparator" colspan="2"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga4c335f86ff2a05daf8dd8aedc628aa59">mlx::core::roll</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &a, const std::vector< int > &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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga6c6fe942a8f7c573a1cb2342566f74d2">mlx::core::roll</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &a, const std::vector< int > &shift, const std::vector< int > &axes, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="separator:ga6c6fe942a8f7c573a1cb2342566f74d2"><td class="memSeparator" colspan="2"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga5e9833b1b7a1c636db6f7ad9809ce902">mlx::core::roll</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &a, int shift, const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &axes, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="separator:ga5e9833b1b7a1c636db6f7ad9809ce902"><td class="memSeparator" colspan="2"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga79137f90bc44ac9e35f408c012701df9">mlx::core::roll</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &a, const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &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"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga9d76930fb567a7d459ff96fb851abe36">mlx::core::roll</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &a, const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &shift, const std::vector< int > &axes, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
|
||||
<tr class="separator:ga9d76930fb567a7d459ff96fb851abe36"><td class="memSeparator" colspan="2"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#gaf8913cabeb9fb193ba687aaeb2087764">mlx::core::real</a> (const <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="separator:gaf8913cabeb9fb193ba687aaeb2087764"><td class="memSeparator" colspan="2"> </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> </td><td class="memItemRight" valign="bottom"><a class="el" href="#ga7ff592a64d528f0cf4f3d098465da029">mlx::core::imag</a> (const <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>
|
||||
@@ -2087,8 +2087,8 @@ Functions</h2></td></tr>
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="ga8de80ecef30fc560003d40f61a38b99d" name="ga8de80ecef30fc560003d40f61a38b99d"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga8de80ecef30fc560003d40f61a38b99d">◆ </a></span>as_strided()</h2>
|
||||
<a id="ga6085b03f2662ef2a61de523fd609f3bf" name="ga6085b03f2662ef2a61de523fd609f3bf"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga6085b03f2662ef2a61de523fd609f3bf">◆ </a></span>as_strided()</h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
@@ -2101,12 +2101,12 @@ Functions</h2></td></tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">std::vector< int ></td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
<td class="paramtype"><a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a></td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">std::vector< size_t ></td> <td class="paramname"><span class="paramname"><em>strides</em></span>, </td>
|
||||
<td class="paramtype"><a class="el" href="namespacemlx_1_1core.html#aca6e505f52ea1464d9c2c09f9f445d62">Strides</a></td> <td class="paramname"><span class="paramname"><em>strides</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
@@ -2436,8 +2436,8 @@ Functions</h2></td></tr>
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="gad256e86cc1a6e6b3832e392baa90318d" name="gad256e86cc1a6e6b3832e392baa90318d"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#gad256e86cc1a6e6b3832e392baa90318d">◆ </a></span>broadcast_to()</h2>
|
||||
<a id="ga2fd5891f11593b7f09550e884f969013" name="ga2fd5891f11593b7f09550e884f969013"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga2fd5891f11593b7f09550e884f969013">◆ </a></span>broadcast_to()</h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
@@ -2450,7 +2450,7 @@ Functions</h2></td></tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">const std::vector< int > &</td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
<td class="paramtype">const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &</td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
@@ -3924,8 +3924,8 @@ Functions</h2></td></tr>
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="ga7e85a476530c96ab73b894d1bddacf84" name="ga7e85a476530c96ab73b894d1bddacf84"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga7e85a476530c96ab73b894d1bddacf84">◆ </a></span>full() <span class="overload">[1/4]</span></h2>
|
||||
<a id="ga1cf232308668fe3f4214c8b895ed4aee" name="ga1cf232308668fe3f4214c8b895ed4aee"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga1cf232308668fe3f4214c8b895ed4aee">◆ </a></span>full() <span class="overload">[1/4]</span></h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
@@ -3933,7 +3933,7 @@ Functions</h2></td></tr>
|
||||
<tr>
|
||||
<td class="memname"><a class="el" href="classmlx_1_1core_1_1array.html">array</a> mlx::core::full </td>
|
||||
<td>(</td>
|
||||
<td class="paramtype">std::vector< int ></td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
<td class="paramtype"><a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a></td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
@@ -3957,8 +3957,8 @@ Functions</h2></td></tr>
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="ga9e3b15c3ec6b7922296777c5e1878d52" name="ga9e3b15c3ec6b7922296777c5e1878d52"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga9e3b15c3ec6b7922296777c5e1878d52">◆ </a></span>full() <span class="overload">[2/4]</span></h2>
|
||||
<a id="ga59f6c844cbb173e108c3eeb11801f8c6" name="ga59f6c844cbb173e108c3eeb11801f8c6"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga59f6c844cbb173e108c3eeb11801f8c6">◆ </a></span>full() <span class="overload">[2/4]</span></h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
@@ -3966,7 +3966,7 @@ Functions</h2></td></tr>
|
||||
<tr>
|
||||
<td class="memname"><a class="el" href="classmlx_1_1core_1_1array.html">array</a> mlx::core::full </td>
|
||||
<td>(</td>
|
||||
<td class="paramtype">std::vector< int ></td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
<td class="paramtype"><a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a></td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
@@ -3983,8 +3983,8 @@ Functions</h2></td></tr>
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="gac34ff7f1e217b93e878053aaa5a51e58" name="gac34ff7f1e217b93e878053aaa5a51e58"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#gac34ff7f1e217b93e878053aaa5a51e58">◆ </a></span>full() <span class="overload">[3/4]</span></h2>
|
||||
<a id="gaf073760b7b51fe35932da0d81c531a55" name="gaf073760b7b51fe35932da0d81c531a55"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#gaf073760b7b51fe35932da0d81c531a55">◆ </a></span>full() <span class="overload">[3/4]</span></h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
@@ -3994,7 +3994,7 @@ template<typename T > </div>
|
||||
<tr>
|
||||
<td class="memname"><a class="el" href="classmlx_1_1core_1_1array.html">array</a> mlx::core::full </td>
|
||||
<td>(</td>
|
||||
<td class="paramtype">std::vector< int ></td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
<td class="paramtype"><a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a></td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
@@ -4016,8 +4016,8 @@ template<typename T > </div>
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="ga5e33cf83db1a65c35e688142bd41d35c" name="ga5e33cf83db1a65c35e688142bd41d35c"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga5e33cf83db1a65c35e688142bd41d35c">◆ </a></span>full() <span class="overload">[4/4]</span></h2>
|
||||
<a id="gaf6f2cce92aff9b71756a3cc3c961fd5a" name="gaf6f2cce92aff9b71756a3cc3c961fd5a"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#gaf6f2cce92aff9b71756a3cc3c961fd5a">◆ </a></span>full() <span class="overload">[4/4]</span></h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
@@ -4027,7 +4027,7 @@ template<typename T > </div>
|
||||
<tr>
|
||||
<td class="memname"><a class="el" href="classmlx_1_1core_1_1array.html">array</a> mlx::core::full </td>
|
||||
<td>(</td>
|
||||
<td class="paramtype">std::vector< int ></td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
<td class="paramtype"><a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a></td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
@@ -4044,8 +4044,8 @@ template<typename T > </div>
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="gadb4337ca5d4f88fe9e7c083bc478158b" name="gadb4337ca5d4f88fe9e7c083bc478158b"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#gadb4337ca5d4f88fe9e7c083bc478158b">◆ </a></span>gather() <span class="overload">[1/2]</span></h2>
|
||||
<a id="gafe2bd174c9953ed7f12664f7abaca0e6" name="gafe2bd174c9953ed7f12664f7abaca0e6"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#gafe2bd174c9953ed7f12664f7abaca0e6">◆ </a></span>gather() <span class="overload">[1/2]</span></h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
@@ -4071,7 +4071,7 @@ template<typename T > </div>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">const std::vector< int > &</td> <td class="paramname"><span class="paramname"><em>slice_sizes</em></span>, </td>
|
||||
<td class="paramtype">const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &</td> <td class="paramname"><span class="paramname"><em>slice_sizes</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
@@ -4088,8 +4088,8 @@ template<typename T > </div>
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="gab6e7f655a9ff15350ca5379692f9d444" name="gab6e7f655a9ff15350ca5379692f9d444"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#gab6e7f655a9ff15350ca5379692f9d444">◆ </a></span>gather() <span class="overload">[2/2]</span></h2>
|
||||
<a id="ga8fcc3ad0677c834c36b72d5b2ebba6d0" name="ga8fcc3ad0677c834c36b72d5b2ebba6d0"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga8fcc3ad0677c834c36b72d5b2ebba6d0">◆ </a></span>gather() <span class="overload">[2/2]</span></h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
@@ -4112,7 +4112,7 @@ template<typename T > </div>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">const std::vector< int > &</td> <td class="paramname"><span class="paramname"><em>slice_sizes</em></span>, </td>
|
||||
<td class="paramtype">const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &</td> <td class="paramname"><span class="paramname"><em>slice_sizes</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
@@ -5684,8 +5684,8 @@ template<typename T > </div>
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="ga54eeed455321a54c8e72e16552a978f2" name="ga54eeed455321a54c8e72e16552a978f2"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga54eeed455321a54c8e72e16552a978f2">◆ </a></span>ones() <span class="overload">[1/2]</span></h2>
|
||||
<a id="gae0069146cf8c819b15ba29aa7231a3f0" name="gae0069146cf8c819b15ba29aa7231a3f0"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#gae0069146cf8c819b15ba29aa7231a3f0">◆ </a></span>ones() <span class="overload">[1/2]</span></h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
@@ -5693,7 +5693,7 @@ template<typename T > </div>
|
||||
<tr>
|
||||
<td class="memname"><a class="el" href="classmlx_1_1core_1_1array.html">array</a> mlx::core::ones </td>
|
||||
<td>(</td>
|
||||
<td class="paramtype">const std::vector< int > &</td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
<td class="paramtype">const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &</td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
@@ -5712,8 +5712,8 @@ template<typename T > </div>
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="ga6cf4b5e8580e4436302c519d05897dab" name="ga6cf4b5e8580e4436302c519d05897dab"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga6cf4b5e8580e4436302c519d05897dab">◆ </a></span>ones() <span class="overload">[2/2]</span></h2>
|
||||
<a id="gace4cf016562af58991f9f961170e156f" name="gace4cf016562af58991f9f961170e156f"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#gace4cf016562af58991f9f961170e156f">◆ </a></span>ones() <span class="overload">[2/2]</span></h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
@@ -5724,7 +5724,7 @@ template<typename T > </div>
|
||||
<tr>
|
||||
<td class="memname"><a class="el" href="classmlx_1_1core_1_1array.html">array</a> mlx::core::ones </td>
|
||||
<td>(</td>
|
||||
<td class="paramtype">const std::vector< int > &</td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
<td class="paramtype">const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &</td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
@@ -7383,8 +7383,8 @@ template<typename T > </div>
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="ga7b0e3d46a39d99b1628b422049790082" name="ga7b0e3d46a39d99b1628b422049790082"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga7b0e3d46a39d99b1628b422049790082">◆ </a></span>reshape()</h2>
|
||||
<a id="ga084f03ce2b22258afb7c8b45e17af828" name="ga084f03ce2b22258afb7c8b45e17af828"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga084f03ce2b22258afb7c8b45e17af828">◆ </a></span>reshape()</h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
@@ -7397,7 +7397,7 @@ template<typename T > </div>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">std::vector< int ></td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
<td class="paramtype"><a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a></td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
@@ -7439,8 +7439,8 @@ template<typename T > </div>
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="ga6c6fe942a8f7c573a1cb2342566f74d2" name="ga6c6fe942a8f7c573a1cb2342566f74d2"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga6c6fe942a8f7c573a1cb2342566f74d2">◆ </a></span>roll() <span class="overload">[1/6]</span></h2>
|
||||
<a id="ga9d76930fb567a7d459ff96fb851abe36" name="ga9d76930fb567a7d459ff96fb851abe36"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga9d76930fb567a7d459ff96fb851abe36">◆ </a></span>roll() <span class="overload">[1/6]</span></h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
@@ -7453,7 +7453,7 @@ template<typename T > </div>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">const std::vector< int > &</td> <td class="paramname"><span class="paramname"><em>shift</em></span>, </td>
|
||||
<td class="paramtype">const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &</td> <td class="paramname"><span class="paramname"><em>shift</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
@@ -7470,8 +7470,8 @@ template<typename T > </div>
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="ga4c335f86ff2a05daf8dd8aedc628aa59" name="ga4c335f86ff2a05daf8dd8aedc628aa59"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga4c335f86ff2a05daf8dd8aedc628aa59">◆ </a></span>roll() <span class="overload">[2/6]</span></h2>
|
||||
<a id="ga79137f90bc44ac9e35f408c012701df9" name="ga79137f90bc44ac9e35f408c012701df9"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga79137f90bc44ac9e35f408c012701df9">◆ </a></span>roll() <span class="overload">[2/6]</span></h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
@@ -7484,7 +7484,7 @@ template<typename T > </div>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">const std::vector< int > &</td> <td class="paramname"><span class="paramname"><em>shift</em></span>, </td>
|
||||
<td class="paramtype">const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &</td> <td class="paramname"><span class="paramname"><em>shift</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
@@ -7501,8 +7501,8 @@ template<typename T > </div>
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="gab30fb942bb6ee00bb50da04033f949a9" name="gab30fb942bb6ee00bb50da04033f949a9"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#gab30fb942bb6ee00bb50da04033f949a9">◆ </a></span>roll() <span class="overload">[3/6]</span></h2>
|
||||
<a id="ga5011d1a5735c64e5b91afa56c7e2cc02" name="ga5011d1a5735c64e5b91afa56c7e2cc02"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga5011d1a5735c64e5b91afa56c7e2cc02">◆ </a></span>roll() <span class="overload">[3/6]</span></h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
@@ -7515,7 +7515,7 @@ template<typename T > </div>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">const std::vector< int > &</td> <td class="paramname"><span class="paramname"><em>shift</em></span>, </td>
|
||||
<td class="paramtype">const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &</td> <td class="paramname"><span class="paramname"><em>shift</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
@@ -7527,8 +7527,8 @@ template<typename T > </div>
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="ga665f502ecc96f1f4467556b784abf9ae" name="ga665f502ecc96f1f4467556b784abf9ae"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga665f502ecc96f1f4467556b784abf9ae">◆ </a></span>roll() <span class="overload">[4/6]</span></h2>
|
||||
<a id="ga5e9833b1b7a1c636db6f7ad9809ce902" name="ga5e9833b1b7a1c636db6f7ad9809ce902"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga5e9833b1b7a1c636db6f7ad9809ce902">◆ </a></span>roll() <span class="overload">[4/6]</span></h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
@@ -7546,7 +7546,7 @@ template<typename T > </div>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">const std::vector< int > &</td> <td class="paramname"><span class="paramname"><em>axes</em></span>, </td>
|
||||
<td class="paramtype">const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &</td> <td class="paramname"><span class="paramname"><em>axes</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
@@ -7779,15 +7779,15 @@ template<typename T > </div>
|
||||
<p>The parameters <code>indices</code> and <code>axes</code> determine the locations of <code>a</code> that are updated with the values in <code>updates</code>. Assuming 1-d <code>indices</code> for simplicity, <code>indices[i]</code> are the indices on axis <code>axes[i]</code> to which the values in <code>updates</code> will be applied. Note each array in <code>indices</code> is assigned to a corresponding axis and hence <code>indices.size() == axes.size()</code>. If an index/axis pair is not provided then indices along that axis are assumed to be zero.</p>
|
||||
<p>Note the rank of <code>updates</code> must be equal to the sum of the rank of the broadcasted <code>indices</code> and the rank of <code>a</code>. In other words, assuming the arrays in <code>indices</code> have the same shape, <code>updates.ndim() == indices[0].ndim() + a.ndim()</code>. The leading dimensions of <code>updates</code> correspond to the indices, and the remaining <code>a.ndim()</code> dimensions are the values that will be applied to the given location in <code>a</code>.</p>
|
||||
<p>For example:</p>
|
||||
<div class="fragment"><div class="line"><span class="keyword">auto</span> in = <a class="code hl_function" href="#ga34573eb2ead179839579142f3aedb3b8">zeros</a>({4, 4}, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>);</div>
|
||||
<div class="fragment"><div class="line"><span class="keyword">auto</span> in = <a class="code hl_function" href="#gae2cace3b388cec4e520659a91879e1c1">zeros</a>({4, 4}, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>);</div>
|
||||
<div class="line"><span class="keyword">auto</span> indices = array({2});</div>
|
||||
<div class="line"><span class="keyword">auto</span> updates = <a class="code hl_function" href="#ga7b0e3d46a39d99b1628b422049790082">reshape</a>(<a class="code hl_function" href="namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b">arange</a>(1, 3, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>), {1, 1, 2});</div>
|
||||
<div class="line"><span class="keyword">auto</span> updates = <a class="code hl_function" href="#ga084f03ce2b22258afb7c8b45e17af828">reshape</a>(<a class="code hl_function" href="namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b">arange</a>(1, 3, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>), {1, 1, 2});</div>
|
||||
<div class="line">std::vector<int> axes{0};</div>
|
||||
<div class="line"> </div>
|
||||
<div class="line"><span class="keyword">auto</span> out = <a class="code hl_function" href="#gad438be8f90bae9d37c6853b8f4225d61">scatter</a>(in, {indices}, updates, axes);</div>
|
||||
<div class="ttc" id="agroup__ops_html_ga34573eb2ead179839579142f3aedb3b8"><div class="ttname"><a href="#ga34573eb2ead179839579142f3aedb3b8">mlx::core::zeros</a></div><div class="ttdeci">array zeros(const std::vector< int > &shape, Dtype dtype, StreamOrDevice s={})</div><div class="ttdoc">Fill an array of the given shape with zeros.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_ga7b0e3d46a39d99b1628b422049790082"><div class="ttname"><a href="#ga7b0e3d46a39d99b1628b422049790082">mlx::core::reshape</a></div><div class="ttdeci">array reshape(const array &a, std::vector< int > shape, StreamOrDevice s={})</div><div class="ttdoc">Reshape an array to the given shape.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_ga084f03ce2b22258afb7c8b45e17af828"><div class="ttname"><a href="#ga084f03ce2b22258afb7c8b45e17af828">mlx::core::reshape</a></div><div class="ttdeci">array reshape(const array &a, Shape shape, StreamOrDevice s={})</div><div class="ttdoc">Reshape an array to the given shape.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_gad438be8f90bae9d37c6853b8f4225d61"><div class="ttname"><a href="#gad438be8f90bae9d37c6853b8f4225d61">mlx::core::scatter</a></div><div class="ttdeci">array scatter(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})</div><div class="ttdoc">Scatter updates to the given indices.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_gae2cace3b388cec4e520659a91879e1c1"><div class="ttname"><a href="#gae2cace3b388cec4e520659a91879e1c1">mlx::core::zeros</a></div><div class="ttdeci">array zeros(const Shape &shape, Dtype dtype, StreamOrDevice s={})</div><div class="ttdoc">Fill an array of the given shape with zeros.</div></div>
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_a369aa886219b83cf219e7a7862ce260b"><div class="ttname"><a href="namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b">mlx::core::arange</a></div><div class="ttdeci">void arange(const std::vector< array > &inputs, array &out, double start, double step)</div><div class="ttdef"><b>Definition</b> arange.h:24</div></div>
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_a6894543b340321193dfb8052c438a319"><div class="ttname"><a href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">mlx::core::float32</a></div><div class="ttdeci">constexpr Dtype float32</div><div class="ttdef"><b>Definition</b> dtype.h:80</div></div>
|
||||
</div><!-- fragment --><p>will produce:</p>
|
||||
@@ -7795,12 +7795,12 @@ template<typename T > </div>
|
||||
<div class="line"> [0, 0, 0, 0],</div>
|
||||
<div class="line"> [1, 2, 0, 0],</div>
|
||||
<div class="line"> [0, 0, 0, 0]], dtype=<a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>)</div>
|
||||
<div class="ttc" id="aclassmlx_1_1core_1_1array_html"><div class="ttname"><a href="classmlx_1_1core_1_1array.html">mlx::core::array</a></div><div class="ttdef"><b>Definition</b> array.h:20</div></div>
|
||||
<div class="ttc" id="aclassmlx_1_1core_1_1array_html"><div class="ttname"><a href="classmlx_1_1core_1_1array.html">mlx::core::array</a></div><div class="ttdef"><b>Definition</b> array.h:23</div></div>
|
||||
</div><!-- fragment --><p>This scatters the two-element row vector <code>[1, 2]</code> starting at the <code>(2, 0)</code> position of <code>a</code>.</p>
|
||||
<p>Adding another element to <code>indices</code> will scatter into another location of <code>a</code>. We also have to add an another update for the new index:</p>
|
||||
<div class="fragment"><div class="line"><span class="keyword">auto</span> in = <a class="code hl_function" href="#ga34573eb2ead179839579142f3aedb3b8">zeros</a>({4, 4}, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>);</div>
|
||||
<div class="fragment"><div class="line"><span class="keyword">auto</span> in = <a class="code hl_function" href="#gae2cace3b388cec4e520659a91879e1c1">zeros</a>({4, 4}, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>);</div>
|
||||
<div class="line"><span class="keyword">auto</span> indices = array({2, 0});</div>
|
||||
<div class="line"><span class="keyword">auto</span> updates = <a class="code hl_function" href="#ga7b0e3d46a39d99b1628b422049790082">reshape</a>(<a class="code hl_function" href="namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b">arange</a>(1, 5, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>), {2, 1, 2});</div>
|
||||
<div class="line"><span class="keyword">auto</span> updates = <a class="code hl_function" href="#ga084f03ce2b22258afb7c8b45e17af828">reshape</a>(<a class="code hl_function" href="namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b">arange</a>(1, 5, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>), {2, 1, 2});</div>
|
||||
<div class="line">std::vector<int> axes{0};</div>
|
||||
<div class="line"> </div>
|
||||
<div class="line"><span class="keyword">auto</span> out = <a class="code hl_function" href="#gad438be8f90bae9d37c6853b8f4225d61">scatter</a>(in, {indices}, updates, axes):</div>
|
||||
@@ -7810,9 +7810,9 @@ template<typename T > </div>
|
||||
<div class="line"> [1, 2, 0, 0],</div>
|
||||
<div class="line"> [0, 0, 0, 0]], dtype=<a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>)</div>
|
||||
</div><!-- fragment --><p>To control the scatter location on an additional axis, add another index array to <code>indices</code> and another axis to <code>axes</code>:</p>
|
||||
<div class="fragment"><div class="line"><span class="keyword">auto</span> in = <a class="code hl_function" href="#ga34573eb2ead179839579142f3aedb3b8">zeros</a>({4, 4}, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>);</div>
|
||||
<div class="fragment"><div class="line"><span class="keyword">auto</span> in = <a class="code hl_function" href="#gae2cace3b388cec4e520659a91879e1c1">zeros</a>({4, 4}, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>);</div>
|
||||
<div class="line"><span class="keyword">auto</span> indices = std::vector{array({2, 0}), array({1, 2})};</div>
|
||||
<div class="line"><span class="keyword">auto</span> updates = <a class="code hl_function" href="#ga7b0e3d46a39d99b1628b422049790082">reshape</a>(<a class="code hl_function" href="namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b">arange</a>(1, 5, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>), {2, 1, 2});</div>
|
||||
<div class="line"><span class="keyword">auto</span> updates = <a class="code hl_function" href="#ga084f03ce2b22258afb7c8b45e17af828">reshape</a>(<a class="code hl_function" href="namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b">arange</a>(1, 5, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>), {2, 1, 2});</div>
|
||||
<div class="line">std::vector<int> axes{0, 1};</div>
|
||||
<div class="line"> </div>
|
||||
<div class="line"><span class="keyword">auto</span> out = <a class="code hl_function" href="#gad438be8f90bae9d37c6853b8f4225d61">scatter</a>(in, indices, updates, axes);</div>
|
||||
@@ -8249,8 +8249,8 @@ template<typename T > </div>
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="gad66135407dbb41b3c5d2cdfd51226c21" name="gad66135407dbb41b3c5d2cdfd51226c21"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#gad66135407dbb41b3c5d2cdfd51226c21">◆ </a></span>slice() <span class="overload">[1/2]</span></h2>
|
||||
<a id="ga29718cd5005dbcde0396b6fd65cc041d" name="ga29718cd5005dbcde0396b6fd65cc041d"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga29718cd5005dbcde0396b6fd65cc041d">◆ </a></span>slice() <span class="overload">[1/2]</span></h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
@@ -8263,17 +8263,17 @@ template<typename T > </div>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">std::vector< int ></td> <td class="paramname"><span class="paramname"><em>start</em></span>, </td>
|
||||
<td class="paramtype"><a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a></td> <td class="paramname"><span class="paramname"><em>start</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">std::vector< int ></td> <td class="paramname"><span class="paramname"><em>stop</em></span>, </td>
|
||||
<td class="paramtype"><a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a></td> <td class="paramname"><span class="paramname"><em>stop</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">std::vector< int ></td> <td class="paramname"><span class="paramname"><em>strides</em></span>, </td>
|
||||
<td class="paramtype"><a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a></td> <td class="paramname"><span class="paramname"><em>strides</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
@@ -8287,8 +8287,8 @@ template<typename T > </div>
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="gaa97ce866c5e38b92b093e9321affcc57" name="gaa97ce866c5e38b92b093e9321affcc57"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#gaa97ce866c5e38b92b093e9321affcc57">◆ </a></span>slice() <span class="overload">[2/2]</span></h2>
|
||||
<a id="gaec56dcb94d5e7f7b885fb60b4bf4aa9d" name="gaec56dcb94d5e7f7b885fb60b4bf4aa9d"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#gaec56dcb94d5e7f7b885fb60b4bf4aa9d">◆ </a></span>slice() <span class="overload">[2/2]</span></h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
@@ -8301,12 +8301,12 @@ template<typename T > </div>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">std::vector< int ></td> <td class="paramname"><span class="paramname"><em>start</em></span>, </td>
|
||||
<td class="paramtype"><a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a></td> <td class="paramname"><span class="paramname"><em>start</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">std::vector< int ></td> <td class="paramname"><span class="paramname"><em>stop</em></span>, </td>
|
||||
<td class="paramtype"><a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a></td> <td class="paramname"><span class="paramname"><em>stop</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
@@ -8320,8 +8320,8 @@ template<typename T > </div>
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="ga3875660e4ce2c8add8bfcf8144078708" name="ga3875660e4ce2c8add8bfcf8144078708"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga3875660e4ce2c8add8bfcf8144078708">◆ </a></span>slice_update() <span class="overload">[1/2]</span></h2>
|
||||
<a id="ga7b7e786985d27789aaed20bb2f9509be" name="ga7b7e786985d27789aaed20bb2f9509be"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga7b7e786985d27789aaed20bb2f9509be">◆ </a></span>slice_update() <span class="overload">[1/2]</span></h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
@@ -8339,17 +8339,17 @@ template<typename T > </div>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">std::vector< int ></td> <td class="paramname"><span class="paramname"><em>start</em></span>, </td>
|
||||
<td class="paramtype"><a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a></td> <td class="paramname"><span class="paramname"><em>start</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">std::vector< int ></td> <td class="paramname"><span class="paramname"><em>stop</em></span>, </td>
|
||||
<td class="paramtype"><a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a></td> <td class="paramname"><span class="paramname"><em>stop</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">std::vector< int ></td> <td class="paramname"><span class="paramname"><em>strides</em></span>, </td>
|
||||
<td class="paramtype"><a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a></td> <td class="paramname"><span class="paramname"><em>strides</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
@@ -8363,8 +8363,8 @@ template<typename T > </div>
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="ga03ffbbb4d989a463ef43f41ebf7eabef" name="ga03ffbbb4d989a463ef43f41ebf7eabef"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga03ffbbb4d989a463ef43f41ebf7eabef">◆ </a></span>slice_update() <span class="overload">[2/2]</span></h2>
|
||||
<a id="ga3b7c6136ae5a38dd6457b65c1833aa67" name="ga3b7c6136ae5a38dd6457b65c1833aa67"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga3b7c6136ae5a38dd6457b65c1833aa67">◆ </a></span>slice_update() <span class="overload">[2/2]</span></h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
@@ -8382,12 +8382,12 @@ template<typename T > </div>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">std::vector< int ></td> <td class="paramname"><span class="paramname"><em>start</em></span>, </td>
|
||||
<td class="paramtype"><a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a></td> <td class="paramname"><span class="paramname"><em>start</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">std::vector< int ></td> <td class="paramname"><span class="paramname"><em>stop</em></span>, </td>
|
||||
<td class="paramtype"><a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a></td> <td class="paramname"><span class="paramname"><em>stop</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
@@ -10046,8 +10046,8 @@ template<typename T > </div>
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="ga34573eb2ead179839579142f3aedb3b8" name="ga34573eb2ead179839579142f3aedb3b8"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga34573eb2ead179839579142f3aedb3b8">◆ </a></span>zeros() <span class="overload">[1/2]</span></h2>
|
||||
<a id="gae2cace3b388cec4e520659a91879e1c1" name="gae2cace3b388cec4e520659a91879e1c1"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#gae2cace3b388cec4e520659a91879e1c1">◆ </a></span>zeros() <span class="overload">[1/2]</span></h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
@@ -10055,7 +10055,7 @@ template<typename T > </div>
|
||||
<tr>
|
||||
<td class="memname"><a class="el" href="classmlx_1_1core_1_1array.html">array</a> mlx::core::zeros </td>
|
||||
<td>(</td>
|
||||
<td class="paramtype">const std::vector< int > &</td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
<td class="paramtype">const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &</td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
@@ -10074,8 +10074,8 @@ template<typename T > </div>
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="ga7c1487f8a830c9df75a961b3a24527ee" name="ga7c1487f8a830c9df75a961b3a24527ee"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ga7c1487f8a830c9df75a961b3a24527ee">◆ </a></span>zeros() <span class="overload">[2/2]</span></h2>
|
||||
<a id="gac8aa722f5e798819b7091693173f1f36" name="gac8aa722f5e798819b7091693173f1f36"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#gac8aa722f5e798819b7091693173f1f36">◆ </a></span>zeros() <span class="overload">[2/2]</span></h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
@@ -10086,7 +10086,7 @@ template<typename T > </div>
|
||||
<tr>
|
||||
<td class="memname"><a class="el" href="classmlx_1_1core_1_1array.html">array</a> mlx::core::zeros </td>
|
||||
<td>(</td>
|
||||
<td class="paramtype">const std::vector< int > &</td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
<td class="paramtype">const <a class="el" href="namespacemlx_1_1core.html#a9be5b57d393e66169279cf521813af9b">Shape</a> &</td> <td class="paramname"><span class="paramname"><em>shape</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
|
Reference in New Issue
Block a user