mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-26 02:33:21 +08:00
1572 lines
366 KiB
HTML
1572 lines
366 KiB
HTML
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "https://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
|
|
<html xmlns="http://www.w3.org/1999/xhtml" lang="en-US">
|
|
<head>
|
|
<meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/>
|
|
<meta http-equiv="X-UA-Compatible" content="IE=11"/>
|
|
<meta name="generator" content="Doxygen 1.12.0"/>
|
|
<meta name="viewport" content="width=device-width, initial-scale=1"/>
|
|
<title>MLX: mlx/ops.h Source File</title>
|
|
<link href="tabs.css" rel="stylesheet" type="text/css"/>
|
|
<script type="text/javascript" src="jquery.js"></script>
|
|
<script type="text/javascript" src="dynsections.js"></script>
|
|
<script type="text/javascript" src="clipboard.js"></script>
|
|
<link href="navtree.css" rel="stylesheet" type="text/css"/>
|
|
<script type="text/javascript" src="resize.js"></script>
|
|
<script type="text/javascript" src="cookie.js"></script>
|
|
<link href="search/search.css" rel="stylesheet" type="text/css"/>
|
|
<script type="text/javascript" src="search/searchdata.js"></script>
|
|
<script type="text/javascript" src="search/search.js"></script>
|
|
<link href="doxygen.css" rel="stylesheet" type="text/css" />
|
|
</head>
|
|
<body>
|
|
<div id="top"><!-- do not remove this div, it is closed by doxygen! -->
|
|
<div id="titlearea">
|
|
<table cellspacing="0" cellpadding="0">
|
|
<tbody>
|
|
<tr id="projectrow">
|
|
<td id="projectalign">
|
|
<div id="projectname">MLX
|
|
</div>
|
|
</td>
|
|
</tr>
|
|
</tbody>
|
|
</table>
|
|
</div>
|
|
<!-- end header part -->
|
|
<!-- Generated by Doxygen 1.12.0 -->
|
|
<script type="text/javascript">
|
|
/* @license magnet:?xt=urn:btih:d3d9a9a6595521f9666a5e94cc830dab83b65699&dn=expat.txt MIT */
|
|
var searchBox = new SearchBox("searchBox", "search/",'.html');
|
|
/* @license-end */
|
|
</script>
|
|
<script type="text/javascript">
|
|
/* @license magnet:?xt=urn:btih:d3d9a9a6595521f9666a5e94cc830dab83b65699&dn=expat.txt MIT */
|
|
$(function() { codefold.init(0); });
|
|
/* @license-end */
|
|
</script>
|
|
<script type="text/javascript" src="menudata.js"></script>
|
|
<script type="text/javascript" src="menu.js"></script>
|
|
<script type="text/javascript">
|
|
/* @license magnet:?xt=urn:btih:d3d9a9a6595521f9666a5e94cc830dab83b65699&dn=expat.txt MIT */
|
|
$(function() {
|
|
initMenu('',true,false,'search.php','Search',false);
|
|
$(function() { init_search(); });
|
|
});
|
|
/* @license-end */
|
|
</script>
|
|
<div id="main-nav"></div>
|
|
<script type="text/javascript">
|
|
/* @license magnet:?xt=urn:btih:d3d9a9a6595521f9666a5e94cc830dab83b65699&dn=expat.txt MIT */
|
|
$(function(){ initResizable(false); });
|
|
/* @license-end */
|
|
</script>
|
|
<!-- window showing the filter options -->
|
|
<div id="MSearchSelectWindow"
|
|
onmouseover="return searchBox.OnSearchSelectShow()"
|
|
onmouseout="return searchBox.OnSearchSelectHide()"
|
|
onkeydown="return searchBox.OnSearchSelectKey(event)">
|
|
</div>
|
|
|
|
<!-- iframe showing the search results (closed by default) -->
|
|
<div id="MSearchResultsWindow">
|
|
<div id="MSearchResults">
|
|
<div class="SRPage">
|
|
<div id="SRIndex">
|
|
<div id="SRResults"></div>
|
|
<div class="SRStatus" id="Loading">Loading...</div>
|
|
<div class="SRStatus" id="Searching">Searching...</div>
|
|
<div class="SRStatus" id="NoMatches">No Matches</div>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
|
|
<div id="nav-path" class="navpath">
|
|
<ul>
|
|
<li class="navelem"><a class="el" href="dir_938ab0ecf10b8b860ff766c820f665fd.html">mlx</a></li> </ul>
|
|
</div>
|
|
</div><!-- top -->
|
|
<div id="doc-content">
|
|
<div class="header">
|
|
<div class="headertitle"><div class="title">ops.h</div></div>
|
|
</div><!--header-->
|
|
<div class="contents">
|
|
<a href="ops_8h.html">Go to the documentation of this file.</a><div class="fragment"><div class="line"><a id="l00001" name="l00001"></a><span class="lineno"> 1</span><span class="comment">// Copyright © 2023-2024 Apple Inc.</span></div>
|
|
<div class="line"><a id="l00002" name="l00002"></a><span class="lineno"> 2</span> </div>
|
|
<div class="line"><a id="l00003" name="l00003"></a><span class="lineno"> 3</span><span class="preprocessor">#pragma once</span></div>
|
|
<div class="line"><a id="l00004" name="l00004"></a><span class="lineno"> 4</span> </div>
|
|
<div class="line"><a id="l00005" name="l00005"></a><span class="lineno"> 5</span><span class="preprocessor">#include <optional></span></div>
|
|
<div class="line"><a id="l00006" name="l00006"></a><span class="lineno"> 6</span> </div>
|
|
<div class="line"><a id="l00007" name="l00007"></a><span class="lineno"> 7</span><span class="preprocessor">#include "<a class="code" href="array_8h.html">mlx/array.h</a>"</span></div>
|
|
<div class="line"><a id="l00008" name="l00008"></a><span class="lineno"> 8</span><span class="preprocessor">#include "<a class="code" href="device_8h.html">mlx/device.h</a>"</span></div>
|
|
<div class="line"><a id="l00009" name="l00009"></a><span class="lineno"> 9</span><span class="preprocessor">#include "<a class="code" href="stream_8h.html">mlx/stream.h</a>"</span></div>
|
|
<div class="line"><a id="l00010" name="l00010"></a><span class="lineno"> 10</span><span class="preprocessor">#include "<a class="code" href="utils_8h.html">mlx/utils.h</a>"</span></div>
|
|
<div class="line"><a id="l00011" name="l00011"></a><span class="lineno"> 11</span> </div>
|
|
<div class="line"><a id="l00012" name="l00012"></a><span class="lineno"> 12</span><span class="keyword">namespace </span><a class="code hl_namespace" href="namespacemlx_1_1core.html">mlx::core</a> {</div>
|
|
<div class="line"><a id="l00013" name="l00013"></a><span class="lineno"> 13</span> </div>
|
|
<div class="line"><a id="l00022" name="l00022"></a><span class="lineno"><a class="line" href="group__ops.html#ga7ca088b8090b9f84f2e08345cf3f835a"> 22</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b">arange</a>(</div>
|
|
<div class="line"><a id="l00023" name="l00023"></a><span class="lineno"> 23</span> <span class="keywordtype">double</span> start,</div>
|
|
<div class="line"><a id="l00024" name="l00024"></a><span class="lineno"> 24</span> <span class="keywordtype">double</span> stop,</div>
|
|
<div class="line"><a id="l00025" name="l00025"></a><span class="lineno"> 25</span> <span class="keywordtype">double</span> step,</div>
|
|
<div class="line"><a id="l00026" name="l00026"></a><span class="lineno"> 26</span> <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype,</div>
|
|
<div class="line"><a id="l00027" name="l00027"></a><span class="lineno"> 27</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00028" name="l00028"></a><span class="lineno"><a class="line" href="group__ops.html#ga4c36b841dc5cba391dad029be5a0ad98"> 28</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b">arange</a>(<span class="keywordtype">double</span> start, <span class="keywordtype">double</span> stop, <span class="keywordtype">double</span> step, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00029" name="l00029"></a><span class="lineno"><a class="line" href="group__ops.html#ga8d7cf9eb15e2daf1469058907e8abc85"> 29</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b">arange</a>(<span class="keywordtype">double</span> start, <span class="keywordtype">double</span> stop, <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00030" name="l00030"></a><span class="lineno"><a class="line" href="group__ops.html#ga74566a14e69ba6a25f5a35e7ade5c282"> 30</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b">arange</a>(<span class="keywordtype">double</span> start, <span class="keywordtype">double</span> stop, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00031" name="l00031"></a><span class="lineno"><a class="line" href="group__ops.html#ga345aa27af3dae3646b8b4b1068e89a3e"> 31</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b">arange</a>(<span class="keywordtype">double</span> stop, <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00032" name="l00032"></a><span class="lineno"><a class="line" href="group__ops.html#gaae179075d0fe23f4bd53fdf8c41f4c70"> 32</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b">arange</a>(<span class="keywordtype">double</span> stop, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00033" name="l00033"></a><span class="lineno"> 33</span> </div>
|
|
<div class="line"><a id="l00034" name="l00034"></a><span class="lineno"><a class="line" href="group__ops.html#ga6b945f513077c2978afc1a952c884860"> 34</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b">arange</a>(<span class="keywordtype">int</span> start, <span class="keywordtype">int</span> stop, <span class="keywordtype">int</span> step, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00035" name="l00035"></a><span class="lineno"><a class="line" href="group__ops.html#ga1c39fcc6eaa1c1867735c7f849d708d6"> 35</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b">arange</a>(<span class="keywordtype">int</span> start, <span class="keywordtype">int</span> stop, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00036" name="l00036"></a><span class="lineno"><a class="line" href="group__ops.html#gafe6e4580452c873cac294f16129e633f"> 36</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b">arange</a>(<span class="keywordtype">int</span> stop, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00037" name="l00037"></a><span class="lineno"> 37</span> </div>
|
|
<div class="line"><a id="l00039" name="l00039"></a><span class="lineno"><a class="line" href="group__ops.html#ga968bcabed902311dcfbd903b0fb886ec"> 39</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga968bcabed902311dcfbd903b0fb886ec">linspace</a>(</div>
|
|
<div class="line"><a id="l00040" name="l00040"></a><span class="lineno"> 40</span> <span class="keywordtype">double</span> start,</div>
|
|
<div class="line"><a id="l00041" name="l00041"></a><span class="lineno"> 41</span> <span class="keywordtype">double</span> stop,</div>
|
|
<div class="line"><a id="l00042" name="l00042"></a><span class="lineno"> 42</span> <span class="keywordtype">int</span> num = 50,</div>
|
|
<div class="line"><a id="l00043" name="l00043"></a><span class="lineno"> 43</span> <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype = <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>,</div>
|
|
<div class="line"><a id="l00044" name="l00044"></a><span class="lineno"> 44</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00045" name="l00045"></a><span class="lineno"> 45</span> </div>
|
|
<div class="line"><a id="l00047" name="l00047"></a><span class="lineno"><a class="line" href="group__ops.html#ga0e58c24fc5668e5a521e5b45e8370a62"> 47</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga0e58c24fc5668e5a521e5b45e8370a62">astype</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> a, <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00048" name="l00048"></a><span class="lineno"> 48</span> </div>
|
|
<div class="line"><a id="l00050" name="l00050"></a><span class="lineno"><a class="line" href="group__ops.html#ga8de80ecef30fc560003d40f61a38b99d"> 50</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga8de80ecef30fc560003d40f61a38b99d">as_strided</a>(</div>
|
|
<div class="line"><a id="l00051" name="l00051"></a><span class="lineno"> 51</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> a,</div>
|
|
<div class="line"><a id="l00052" name="l00052"></a><span class="lineno"> 52</span> std::vector<int> shape,</div>
|
|
<div class="line"><a id="l00053" name="l00053"></a><span class="lineno"> 53</span> std::vector<size_t> strides,</div>
|
|
<div class="line"><a id="l00054" name="l00054"></a><span class="lineno"> 54</span> <span class="keywordtype">size_t</span> offset,</div>
|
|
<div class="line"><a id="l00055" name="l00055"></a><span class="lineno"> 55</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00056" name="l00056"></a><span class="lineno"> 56</span> </div>
|
|
<div class="line"><a id="l00058" name="l00058"></a><span class="lineno"><a class="line" href="group__ops.html#gae306e93af12f774bd80bad6c231b09d6"> 58</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="namespacemlx_1_1core.html#a479648542a2bea151b947b18f0e79dd2">copy</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00059" name="l00059"></a><span class="lineno"> 59</span> </div>
|
|
<div class="line"><a id="l00061" name="l00061"></a><span class="lineno"><a class="line" href="group__ops.html#ga7e85a476530c96ab73b894d1bddacf84"> 61</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7e85a476530c96ab73b894d1bddacf84">full</a>(</div>
|
|
<div class="line"><a id="l00062" name="l00062"></a><span class="lineno"> 62</span> std::vector<int> shape,</div>
|
|
<div class="line"><a id="l00063" name="l00063"></a><span class="lineno"> 63</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> vals,</div>
|
|
<div class="line"><a id="l00064" name="l00064"></a><span class="lineno"> 64</span> <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype,</div>
|
|
<div class="line"><a id="l00065" name="l00065"></a><span class="lineno"> 65</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00066" name="l00066"></a><span class="lineno"><a class="line" href="group__ops.html#ga9e3b15c3ec6b7922296777c5e1878d52"> 66</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7e85a476530c96ab73b894d1bddacf84">full</a>(std::vector<int> shape, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> vals, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00067" name="l00067"></a><span class="lineno"> 67</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00068" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00068" name="l00068"></a><span class="lineno"><a class="line" href="group__ops.html#gac34ff7f1e217b93e878053aaa5a51e58"> 68</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7e85a476530c96ab73b894d1bddacf84">full</a>(std::vector<int> shape, T val, <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00069" name="l00069"></a><span class="lineno"> 69</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga7e85a476530c96ab73b894d1bddacf84">full</a>(std::move(shape), array(val, dtype), <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00070" name="l00070"></a><span class="lineno"> 70</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00071" name="l00071"></a><span class="lineno"> 71</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00072" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00072" name="l00072"></a><span class="lineno"><a class="line" href="group__ops.html#ga5e33cf83db1a65c35e688142bd41d35c"> 72</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7e85a476530c96ab73b894d1bddacf84">full</a>(std::vector<int> shape, T val, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00073" name="l00073"></a><span class="lineno"> 73</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga7e85a476530c96ab73b894d1bddacf84">full</a>(std::move(shape), array(val), <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00074" name="l00074"></a><span class="lineno"> 74</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00075" name="l00075"></a><span class="lineno"> 75</span> </div>
|
|
<div class="line"><a id="l00077" name="l00077"></a><span class="lineno"><a class="line" href="group__ops.html#ga34573eb2ead179839579142f3aedb3b8"> 77</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga34573eb2ead179839579142f3aedb3b8">zeros</a>(<span class="keyword">const</span> std::vector<int>& shape, <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00078" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00078" name="l00078"></a><span class="lineno"><a class="line" href="group__ops.html#ga7c1487f8a830c9df75a961b3a24527ee"> 78</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga34573eb2ead179839579142f3aedb3b8">zeros</a>(<span class="keyword">const</span> std::vector<int>& shape, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00079" name="l00079"></a><span class="lineno"> 79</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga34573eb2ead179839579142f3aedb3b8">zeros</a>(shape, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>, s);</div>
|
|
<div class="line"><a id="l00080" name="l00080"></a><span class="lineno"> 80</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00081" name="l00081"></a><span class="lineno"><a class="line" href="group__ops.html#gafbb857094d784b38c78683a091ffdbde"> 81</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gafbb857094d784b38c78683a091ffdbde">zeros_like</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00082" name="l00082"></a><span class="lineno"> 82</span> </div>
|
|
<div class="line"><a id="l00084" name="l00084"></a><span class="lineno"><a class="line" href="group__ops.html#ga54eeed455321a54c8e72e16552a978f2"> 84</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga54eeed455321a54c8e72e16552a978f2">ones</a>(<span class="keyword">const</span> std::vector<int>& shape, <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00085" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00085" name="l00085"></a><span class="lineno"><a class="line" href="group__ops.html#ga6cf4b5e8580e4436302c519d05897dab"> 85</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga54eeed455321a54c8e72e16552a978f2">ones</a>(<span class="keyword">const</span> std::vector<int>& shape, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00086" name="l00086"></a><span class="lineno"> 86</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga54eeed455321a54c8e72e16552a978f2">ones</a>(shape, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>, s);</div>
|
|
<div class="line"><a id="l00087" name="l00087"></a><span class="lineno"> 87</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00088" name="l00088"></a><span class="lineno"><a class="line" href="group__ops.html#ga94f8d3b1906fee99da9cbe39f7be7d42"> 88</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga94f8d3b1906fee99da9cbe39f7be7d42">ones_like</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00089" name="l00089"></a><span class="lineno"> 89</span> </div>
|
|
<div class="line"><a id="l00092" name="l00092"></a><span class="lineno"><a class="line" href="group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3"> 92</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3">eye</a>(<span class="keywordtype">int</span> n, <span class="keywordtype">int</span> m, <span class="keywordtype">int</span> k, <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00093" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00093" name="l00093"></a><span class="lineno"><a class="line" href="group__ops.html#ga2c9011310a1fa7c82f942f54102c36dd"> 93</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3">eye</a>(<span class="keywordtype">int</span> n, <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00094" name="l00094"></a><span class="lineno"> 94</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3">eye</a>(n, n, 0, dtype, s);</div>
|
|
<div class="line"><a id="l00095" name="l00095"></a><span class="lineno"> 95</span>}</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00096" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00096" name="l00096"></a><span class="lineno"><a class="line" href="group__ops.html#ga61657db78ef35d41112d362c869c25d2"> 96</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3">eye</a>(<span class="keywordtype">int</span> n, <span class="keywordtype">int</span> m, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00097" name="l00097"></a><span class="lineno"> 97</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3">eye</a>(n, m, 0, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>, s);</div>
|
|
<div class="line"><a id="l00098" name="l00098"></a><span class="lineno"> 98</span>}</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00099" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00099" name="l00099"></a><span class="lineno"><a class="line" href="group__ops.html#ga908a15b42834be498a46856c99dfc779"> 99</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3">eye</a>(<span class="keywordtype">int</span> n, <span class="keywordtype">int</span> m, <span class="keywordtype">int</span> k, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00100" name="l00100"></a><span class="lineno"> 100</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3">eye</a>(n, m, k, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>, s);</div>
|
|
<div class="line"><a id="l00101" name="l00101"></a><span class="lineno"> 101</span>}</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00102" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00102" name="l00102"></a><span class="lineno"><a class="line" href="group__ops.html#gab777fcf6d4a89172c69ec3492548dc0f"> 102</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3">eye</a>(<span class="keywordtype">int</span> n, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00103" name="l00103"></a><span class="lineno"> 103</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3">eye</a>(n, n, 0, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>, s);</div>
|
|
<div class="line"><a id="l00104" name="l00104"></a><span class="lineno"> 104</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00105" name="l00105"></a><span class="lineno"> 105</span> </div>
|
|
<div class="line"><a id="l00108" name="l00108"></a><span class="lineno"><a class="line" href="group__ops.html#ga484eaa10d5e19a4ca46d3a9cd9fab600"> 108</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga484eaa10d5e19a4ca46d3a9cd9fab600">identity</a>(<span class="keywordtype">int</span> n, <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00109" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00109" name="l00109"></a><span class="lineno"><a class="line" href="group__ops.html#gad994d65ac6019c26b5ad6c41179d3424"> 109</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga484eaa10d5e19a4ca46d3a9cd9fab600">identity</a>(<span class="keywordtype">int</span> n, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00110" name="l00110"></a><span class="lineno"> 110</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga484eaa10d5e19a4ca46d3a9cd9fab600">identity</a>(n, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>, s);</div>
|
|
<div class="line"><a id="l00111" name="l00111"></a><span class="lineno"> 111</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00112" name="l00112"></a><span class="lineno"> 112</span> </div>
|
|
<div class="line"><a id="l00113" name="l00113"></a><span class="lineno"><a class="line" href="group__ops.html#ga4f3389e5b89e70e862e7d2b40d6c7f78"> 113</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga4f3389e5b89e70e862e7d2b40d6c7f78">tri</a>(<span class="keywordtype">int</span> n, <span class="keywordtype">int</span> m, <span class="keywordtype">int</span> k, <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> type, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00114" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00114" name="l00114"></a><span class="lineno"><a class="line" href="group__ops.html#gac19a1bd6ed6d5c7bc9d258820189dbb5"> 114</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga4f3389e5b89e70e862e7d2b40d6c7f78">tri</a>(<span class="keywordtype">int</span> n, <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> type, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00115" name="l00115"></a><span class="lineno"> 115</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga4f3389e5b89e70e862e7d2b40d6c7f78">tri</a>(n, n, 0, type, s);</div>
|
|
<div class="line"><a id="l00116" name="l00116"></a><span class="lineno"> 116</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00117" name="l00117"></a><span class="lineno"> 117</span> </div>
|
|
<div class="line"><a id="l00118" name="l00118"></a><span class="lineno"><a class="line" href="group__ops.html#ga83e0bb45dc770cf014531d873b78c5a2"> 118</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga83e0bb45dc770cf014531d873b78c5a2">tril</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> x, <span class="keywordtype">int</span> k = 0, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00119" name="l00119"></a><span class="lineno"><a class="line" href="group__ops.html#gaa9df5917876eeb0cb28b7fa81f880412"> 119</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaa9df5917876eeb0cb28b7fa81f880412">triu</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> x, <span class="keywordtype">int</span> k = 0, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00120" name="l00120"></a><span class="lineno"> 120</span> </div>
|
|
<div class="line"><a id="l00122" name="l00122"></a><span class="lineno"><a class="line" href="group__ops.html#ga7b0e3d46a39d99b1628b422049790082"> 122</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7b0e3d46a39d99b1628b422049790082">reshape</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, std::vector<int> shape, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00123" name="l00123"></a><span class="lineno"> 123</span> </div>
|
|
<div class="line"><a id="l00125" name="l00125"></a><span class="lineno"><a class="line" href="group__ops.html#ga50aa98754b412bb57c083f6e3e95061f"> 125</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga50aa98754b412bb57c083f6e3e95061f">flatten</a>(</div>
|
|
<div class="line"><a id="l00126" name="l00126"></a><span class="lineno"> 126</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00127" name="l00127"></a><span class="lineno"> 127</span> <span class="keywordtype">int</span> start_axis,</div>
|
|
<div class="line"><a id="l00128" name="l00128"></a><span class="lineno"> 128</span> <span class="keywordtype">int</span> end_axis = -1,</div>
|
|
<div class="line"><a id="l00129" name="l00129"></a><span class="lineno"> 129</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00130" name="l00130"></a><span class="lineno"> 130</span> </div>
|
|
<div class="line"><a id="l00132" name="l00132"></a><span class="lineno"><a class="line" href="group__ops.html#gaa6adbc9c86f0ab27d8810a02e9e719fd"> 132</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga50aa98754b412bb57c083f6e3e95061f">flatten</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00133" name="l00133"></a><span class="lineno"> 133</span> </div>
|
|
<div class="line"><a id="l00135" name="l00135"></a><span class="lineno"><a class="line" href="group__ops.html#ga872d2c1806e67ce2596b24d056681074"> 135</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga872d2c1806e67ce2596b24d056681074">hadamard_transform</a>(</div>
|
|
<div class="line"><a id="l00136" name="l00136"></a><span class="lineno"> 136</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00137" name="l00137"></a><span class="lineno"> 137</span> std::optional<float> scale = std::nullopt,</div>
|
|
<div class="line"><a id="l00138" name="l00138"></a><span class="lineno"> 138</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00139" name="l00139"></a><span class="lineno"> 139</span> </div>
|
|
<div class="line"><a id="l00141" name="l00141"></a><span class="lineno"><a class="line" href="group__ops.html#ga710daa7ec721bd4d3f326082cb195576"> 141</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga710daa7ec721bd4d3f326082cb195576">squeeze</a>(</div>
|
|
<div class="line"><a id="l00142" name="l00142"></a><span class="lineno"> 142</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00143" name="l00143"></a><span class="lineno"> 143</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00144" name="l00144"></a><span class="lineno"> 144</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00145" name="l00145"></a><span class="lineno"> 145</span> </div>
|
|
<div class="line"><a id="l00147" name="l00147"></a><span class="lineno"><a class="line" href="group__ops.html#ga700dd51b77379a3d2260a55783e8ebf3"> 147</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga710daa7ec721bd4d3f326082cb195576">squeeze</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> axis, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00148" name="l00148"></a><span class="lineno"> 148</span> </div>
|
|
<div class="line"><a id="l00150" name="l00150"></a><span class="lineno"><a class="line" href="group__ops.html#ga58bad3c61fd85b95927a987ba1cf5dad"> 150</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga710daa7ec721bd4d3f326082cb195576">squeeze</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00151" name="l00151"></a><span class="lineno"> 151</span> </div>
|
|
<div class="line"><a id="l00153" name="l00153"></a><span class="lineno"><a class="line" href="group__ops.html#ga717f11149a8c7b4cc3e33bbcc0a97133"> 153</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga717f11149a8c7b4cc3e33bbcc0a97133">expand_dims</a>(</div>
|
|
<div class="line"><a id="l00154" name="l00154"></a><span class="lineno"> 154</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00155" name="l00155"></a><span class="lineno"> 155</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00156" name="l00156"></a><span class="lineno"> 156</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00157" name="l00157"></a><span class="lineno"> 157</span> </div>
|
|
<div class="line"><a id="l00159" name="l00159"></a><span class="lineno"><a class="line" href="group__ops.html#ga7a80adb4a5a36d18b5f234d4b034950a"> 159</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga717f11149a8c7b4cc3e33bbcc0a97133">expand_dims</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> axis, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00160" name="l00160"></a><span class="lineno"> 160</span> </div>
|
|
<div class="line"><a id="l00162" name="l00162"></a><span class="lineno"><a class="line" href="group__ops.html#gad66135407dbb41b3c5d2cdfd51226c21"> 162</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gad66135407dbb41b3c5d2cdfd51226c21">slice</a>(</div>
|
|
<div class="line"><a id="l00163" name="l00163"></a><span class="lineno"> 163</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00164" name="l00164"></a><span class="lineno"> 164</span> std::vector<int> start,</div>
|
|
<div class="line"><a id="l00165" name="l00165"></a><span class="lineno"> 165</span> std::vector<int> stop,</div>
|
|
<div class="line"><a id="l00166" name="l00166"></a><span class="lineno"> 166</span> std::vector<int> strides,</div>
|
|
<div class="line"><a id="l00167" name="l00167"></a><span class="lineno"> 167</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00168" name="l00168"></a><span class="lineno"> 168</span> </div>
|
|
<div class="line"><a id="l00170" name="l00170"></a><span class="lineno"><a class="line" href="group__ops.html#gaa97ce866c5e38b92b093e9321affcc57"> 170</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gad66135407dbb41b3c5d2cdfd51226c21">slice</a>(</div>
|
|
<div class="line"><a id="l00171" name="l00171"></a><span class="lineno"> 171</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00172" name="l00172"></a><span class="lineno"> 172</span> std::vector<int> start,</div>
|
|
<div class="line"><a id="l00173" name="l00173"></a><span class="lineno"> 173</span> std::vector<int> stop,</div>
|
|
<div class="line"><a id="l00174" name="l00174"></a><span class="lineno"> 174</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00175" name="l00175"></a><span class="lineno"> 175</span> </div>
|
|
<div class="line"><a id="l00177" name="l00177"></a><span class="lineno"><a class="line" href="group__ops.html#ga3875660e4ce2c8add8bfcf8144078708"> 177</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga3875660e4ce2c8add8bfcf8144078708">slice_update</a>(</div>
|
|
<div class="line"><a id="l00178" name="l00178"></a><span class="lineno"> 178</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& src,</div>
|
|
<div class="line"><a id="l00179" name="l00179"></a><span class="lineno"> 179</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& update,</div>
|
|
<div class="line"><a id="l00180" name="l00180"></a><span class="lineno"> 180</span> std::vector<int> start,</div>
|
|
<div class="line"><a id="l00181" name="l00181"></a><span class="lineno"> 181</span> std::vector<int> stop,</div>
|
|
<div class="line"><a id="l00182" name="l00182"></a><span class="lineno"> 182</span> std::vector<int> strides,</div>
|
|
<div class="line"><a id="l00183" name="l00183"></a><span class="lineno"> 183</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00184" name="l00184"></a><span class="lineno"> 184</span> </div>
|
|
<div class="line"><a id="l00186" name="l00186"></a><span class="lineno"><a class="line" href="group__ops.html#ga03ffbbb4d989a463ef43f41ebf7eabef"> 186</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga3875660e4ce2c8add8bfcf8144078708">slice_update</a>(</div>
|
|
<div class="line"><a id="l00187" name="l00187"></a><span class="lineno"> 187</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& src,</div>
|
|
<div class="line"><a id="l00188" name="l00188"></a><span class="lineno"> 188</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& update,</div>
|
|
<div class="line"><a id="l00189" name="l00189"></a><span class="lineno"> 189</span> std::vector<int> start,</div>
|
|
<div class="line"><a id="l00190" name="l00190"></a><span class="lineno"> 190</span> std::vector<int> stop,</div>
|
|
<div class="line"><a id="l00191" name="l00191"></a><span class="lineno"> 191</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00192" name="l00192"></a><span class="lineno"> 192</span> </div>
|
|
<div class="line"><a id="l00194" name="l00194"></a><span class="lineno"> 194</span>std::vector<array></div>
|
|
<div class="line"><a id="l00195" name="l00195"></a><span class="lineno"><a class="line" href="group__ops.html#ga7534290bceab5fb3831a05d67bebce7d"> 195</a></span><a class="code hl_function" href="group__ops.html#ga7534290bceab5fb3831a05d67bebce7d">split</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> num_splits, <span class="keywordtype">int</span> axis, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00196" name="l00196"></a><span class="lineno"><a class="line" href="group__ops.html#ga56882d24e5fde59c266774624c892d41"> 196</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#ga7534290bceab5fb3831a05d67bebce7d">split</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> num_splits, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00197" name="l00197"></a><span class="lineno"><a class="line" href="group__ops.html#ga2cfcb1a53924882e30476c9016c5de74"> 197</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#ga7534290bceab5fb3831a05d67bebce7d">split</a>(</div>
|
|
<div class="line"><a id="l00198" name="l00198"></a><span class="lineno"> 198</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00199" name="l00199"></a><span class="lineno"> 199</span> <span class="keyword">const</span> std::vector<int>& indices,</div>
|
|
<div class="line"><a id="l00200" name="l00200"></a><span class="lineno"> 200</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00201" name="l00201"></a><span class="lineno"> 201</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00202" name="l00202"></a><span class="lineno"> 202</span>std::vector<array></div>
|
|
<div class="line"><a id="l00203" name="l00203"></a><span class="lineno"><a class="line" href="group__ops.html#gac324dfa3e26d3a14a35ab7962e36f0e1"> 203</a></span><a class="code hl_function" href="group__ops.html#ga7534290bceab5fb3831a05d67bebce7d">split</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> std::vector<int>& indices, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00204" name="l00204"></a><span class="lineno"> 204</span> </div>
|
|
<div class="line"><a id="l00206" name="l00206"></a><span class="lineno"><a class="line" href="group__ops.html#ga577c911618575314de63d1060656a26e"> 206</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#ga577c911618575314de63d1060656a26e">meshgrid</a>(</div>
|
|
<div class="line"><a id="l00207" name="l00207"></a><span class="lineno"> 207</span> <span class="keyword">const</span> std::vector<array>& arrays,</div>
|
|
<div class="line"><a id="l00208" name="l00208"></a><span class="lineno"> 208</span> <span class="keywordtype">bool</span> sparse = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00209" name="l00209"></a><span class="lineno"> 209</span> std::string indexing = <span class="stringliteral">"xy"</span>,</div>
|
|
<div class="line"><a id="l00210" name="l00210"></a><span class="lineno"> 210</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00211" name="l00211"></a><span class="lineno"> 211</span> </div>
|
|
<div class="line"><a id="l00215" name="l00215"></a><span class="lineno"><a class="line" href="group__ops.html#ga157cd7c23f9b306fee2e1eb2b9bf1dd8"> 215</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga157cd7c23f9b306fee2e1eb2b9bf1dd8">clip</a>(</div>
|
|
<div class="line"><a id="l00216" name="l00216"></a><span class="lineno"> 216</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00217" name="l00217"></a><span class="lineno"> 217</span> <span class="keyword">const</span> std::optional<array>& a_min = std::nullopt,</div>
|
|
<div class="line"><a id="l00218" name="l00218"></a><span class="lineno"> 218</span> <span class="keyword">const</span> std::optional<array>& a_max = std::nullopt,</div>
|
|
<div class="line"><a id="l00219" name="l00219"></a><span class="lineno"> 219</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00220" name="l00220"></a><span class="lineno"> 220</span> </div>
|
|
<div class="line"><a id="l00222" name="l00222"></a><span class="lineno"><a class="line" href="group__ops.html#gabdc36fa65697d0361c8d67495de77129"> 222</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gabdc36fa65697d0361c8d67495de77129">concatenate</a>(</div>
|
|
<div class="line"><a id="l00223" name="l00223"></a><span class="lineno"> 223</span> <span class="keyword">const</span> std::vector<array>& arrays,</div>
|
|
<div class="line"><a id="l00224" name="l00224"></a><span class="lineno"> 224</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00225" name="l00225"></a><span class="lineno"> 225</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00226" name="l00226"></a><span class="lineno"><a class="line" href="group__ops.html#gaa95c34ca3a8877f2c50cb60e7fa312b8"> 226</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gabdc36fa65697d0361c8d67495de77129">concatenate</a>(<span class="keyword">const</span> std::vector<array>& arrays, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00227" name="l00227"></a><span class="lineno"> 227</span> </div>
|
|
<div class="line"><a id="l00229" name="l00229"></a><span class="lineno"><a class="line" href="group__ops.html#gaf8f2ec2b98a4b59eca73d7471df6e032"> 229</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaf8f2ec2b98a4b59eca73d7471df6e032">stack</a>(<span class="keyword">const</span> std::vector<array>& arrays, <span class="keywordtype">int</span> axis, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00230" name="l00230"></a><span class="lineno"><a class="line" href="group__ops.html#ga82216209dce901296fc737fe8efa5c94"> 230</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaf8f2ec2b98a4b59eca73d7471df6e032">stack</a>(<span class="keyword">const</span> std::vector<array>& arrays, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00231" name="l00231"></a><span class="lineno"> 231</span> </div>
|
|
<div class="line"><a id="l00233" name="l00233"></a><span class="lineno"><a class="line" href="group__ops.html#gab49e3a687e826554ed1574186e8ae974"> 233</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gab49e3a687e826554ed1574186e8ae974">repeat</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& arr, <span class="keywordtype">int</span> repeats, <span class="keywordtype">int</span> axis, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00234" name="l00234"></a><span class="lineno"><a class="line" href="group__ops.html#ga4f75f5d5db999f02f43ecbc6dccf3ba6"> 234</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gab49e3a687e826554ed1574186e8ae974">repeat</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& arr, <span class="keywordtype">int</span> repeats, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00235" name="l00235"></a><span class="lineno"> 235</span> </div>
|
|
<div class="line"><a id="l00236" name="l00236"></a><span class="lineno"><a class="line" href="group__ops.html#gab105a57b9a4d84496fe1e4d60e13d361"> 236</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gab105a57b9a4d84496fe1e4d60e13d361">tile</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& arr, std::vector<int> reps, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00237" name="l00237"></a><span class="lineno"> 237</span> </div>
|
|
<div class="line"><a id="l00239" name="l00239"></a><span class="lineno"><a class="line" href="group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b"> 239</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b">transpose</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, std::vector<int> axes, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00240" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00240" name="l00240"></a><span class="lineno"><a class="line" href="group__ops.html#ga260ac332956f3a6bf1dfdb9095c84dc5"> 240</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b">transpose</a>(</div>
|
|
<div class="line"><a id="l00241" name="l00241"></a><span class="lineno"> 241</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00242" name="l00242"></a><span class="lineno"> 242</span> std::initializer_list<int> axes,</div>
|
|
<div class="line"><a id="l00243" name="l00243"></a><span class="lineno"> 243</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00244" name="l00244"></a><span class="lineno"> 244</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b">transpose</a>(a, std::vector<int>(axes), s);</div>
|
|
<div class="line"><a id="l00245" name="l00245"></a><span class="lineno"> 245</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00246" name="l00246"></a><span class="lineno"> 246</span> </div>
|
|
<div class="line"><a id="l00248" name="l00248"></a><span class="lineno"><a class="line" href="group__ops.html#gabc46eed81ab6c6247903e4ec0c4ec1fb"> 248</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gabc46eed81ab6c6247903e4ec0c4ec1fb">swapaxes</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> axis1, <span class="keywordtype">int</span> axis2, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00249" name="l00249"></a><span class="lineno"> 249</span> </div>
|
|
<div class="line"><a id="l00251" name="l00251"></a><span class="lineno"><a class="line" href="group__ops.html#ga24067d10a842db2c9d509ea48135a2c3"> 251</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga24067d10a842db2c9d509ea48135a2c3">moveaxis</a>(</div>
|
|
<div class="line"><a id="l00252" name="l00252"></a><span class="lineno"> 252</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00253" name="l00253"></a><span class="lineno"> 253</span> <span class="keywordtype">int</span> source,</div>
|
|
<div class="line"><a id="l00254" name="l00254"></a><span class="lineno"> 254</span> <span class="keywordtype">int</span> destination,</div>
|
|
<div class="line"><a id="l00255" name="l00255"></a><span class="lineno"> 255</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00256" name="l00256"></a><span class="lineno"> 256</span> </div>
|
|
<div class="line"><a id="l00258" name="l00258"></a><span class="lineno"><a class="line" href="group__ops.html#ga81c2f138ea81d78affd42e9a646cacbc"> 258</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga81c2f138ea81d78affd42e9a646cacbc">pad</a>(</div>
|
|
<div class="line"><a id="l00259" name="l00259"></a><span class="lineno"> 259</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00260" name="l00260"></a><span class="lineno"> 260</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00261" name="l00261"></a><span class="lineno"> 261</span> <span class="keyword">const</span> std::vector<int>& low_pad_size,</div>
|
|
<div class="line"><a id="l00262" name="l00262"></a><span class="lineno"> 262</span> <span class="keyword">const</span> std::vector<int>& high_pad_size,</div>
|
|
<div class="line"><a id="l00263" name="l00263"></a><span class="lineno"> 263</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& pad_value = <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(0),</div>
|
|
<div class="line"><a id="l00264" name="l00264"></a><span class="lineno"> 264</span> <span class="keyword">const</span> std::string mode = <span class="stringliteral">"constant"</span>,</div>
|
|
<div class="line"><a id="l00265" name="l00265"></a><span class="lineno"> 265</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00266" name="l00266"></a><span class="lineno"> 266</span> </div>
|
|
<div class="line"><a id="l00268" name="l00268"></a><span class="lineno"><a class="line" href="group__ops.html#ga8495c9902b0a67f288918cb528d5abeb"> 268</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga81c2f138ea81d78affd42e9a646cacbc">pad</a>(</div>
|
|
<div class="line"><a id="l00269" name="l00269"></a><span class="lineno"> 269</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00270" name="l00270"></a><span class="lineno"> 270</span> <span class="keyword">const</span> std::vector<std::pair<int, int>>& pad_width,</div>
|
|
<div class="line"><a id="l00271" name="l00271"></a><span class="lineno"> 271</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& pad_value = <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(0),</div>
|
|
<div class="line"><a id="l00272" name="l00272"></a><span class="lineno"> 272</span> <span class="keyword">const</span> std::string mode = <span class="stringliteral">"constant"</span>,</div>
|
|
<div class="line"><a id="l00273" name="l00273"></a><span class="lineno"> 273</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00274" name="l00274"></a><span class="lineno"><a class="line" href="group__ops.html#gaeaae2f9637b0b3e2b4c6c0517e285935"> 274</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga81c2f138ea81d78affd42e9a646cacbc">pad</a>(</div>
|
|
<div class="line"><a id="l00275" name="l00275"></a><span class="lineno"> 275</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00276" name="l00276"></a><span class="lineno"> 276</span> <span class="keyword">const</span> std::pair<int, int>& pad_width,</div>
|
|
<div class="line"><a id="l00277" name="l00277"></a><span class="lineno"> 277</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& pad_value = <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(0),</div>
|
|
<div class="line"><a id="l00278" name="l00278"></a><span class="lineno"> 278</span> <span class="keyword">const</span> std::string mode = <span class="stringliteral">"constant"</span>,</div>
|
|
<div class="line"><a id="l00279" name="l00279"></a><span class="lineno"> 279</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00280" name="l00280"></a><span class="lineno"><a class="line" href="group__ops.html#ga5b8684361f57cbfd5e6eedf82a62ded0"> 280</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga81c2f138ea81d78affd42e9a646cacbc">pad</a>(</div>
|
|
<div class="line"><a id="l00281" name="l00281"></a><span class="lineno"> 281</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00282" name="l00282"></a><span class="lineno"> 282</span> <span class="keywordtype">int</span> pad_width,</div>
|
|
<div class="line"><a id="l00283" name="l00283"></a><span class="lineno"> 283</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& pad_value = <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(0),</div>
|
|
<div class="line"><a id="l00284" name="l00284"></a><span class="lineno"> 284</span> <span class="keyword">const</span> std::string mode = <span class="stringliteral">"constant"</span>,</div>
|
|
<div class="line"><a id="l00285" name="l00285"></a><span class="lineno"> 285</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00286" name="l00286"></a><span class="lineno"> 286</span> </div>
|
|
<div class="line"><a id="l00288" name="l00288"></a><span class="lineno"><a class="line" href="group__ops.html#ga68da0176fefbe0c0096783c6fd926c6a"> 288</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b">transpose</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00289" name="l00289"></a><span class="lineno"> 289</span> </div>
|
|
<div class="line"><a id="l00291" name="l00291"></a><span class="lineno"><a class="line" href="group__ops.html#gad256e86cc1a6e6b3832e392baa90318d"> 291</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gad256e86cc1a6e6b3832e392baa90318d">broadcast_to</a>(</div>
|
|
<div class="line"><a id="l00292" name="l00292"></a><span class="lineno"> 292</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00293" name="l00293"></a><span class="lineno"> 293</span> <span class="keyword">const</span> std::vector<int>& shape,</div>
|
|
<div class="line"><a id="l00294" name="l00294"></a><span class="lineno"> 294</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00295" name="l00295"></a><span class="lineno"> 295</span> </div>
|
|
<div class="line"><a id="l00297" name="l00297"></a><span class="lineno"><a class="line" href="group__ops.html#gab783890428b596f715dc7dd2057eae99"> 297</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#gab783890428b596f715dc7dd2057eae99">broadcast_arrays</a>(</div>
|
|
<div class="line"><a id="l00298" name="l00298"></a><span class="lineno"> 298</span> <span class="keyword">const</span> std::vector<array>& inputs,</div>
|
|
<div class="line"><a id="l00299" name="l00299"></a><span class="lineno"> 299</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00300" name="l00300"></a><span class="lineno"> 300</span> </div>
|
|
<div class="line"><a id="l00302" name="l00302"></a><span class="lineno"><a class="line" href="group__ops.html#ga33638dc3a9972dd02be12d0eb85f9bde"> 302</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga33638dc3a9972dd02be12d0eb85f9bde">equal</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00303" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00303" name="l00303"></a><span class="lineno"><a class="line" href="group__ops.html#gaa30cf69f3d22f65615f5e1696dd5703f"> 303</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="namespacemlx_1_1core.html#a937503d72b66c661bf3f5fdcd98ef97c">operator==</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b) {</div>
|
|
<div class="line"><a id="l00304" name="l00304"></a><span class="lineno"> 304</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga33638dc3a9972dd02be12d0eb85f9bde">equal</a>(a, b);</div>
|
|
<div class="line"><a id="l00305" name="l00305"></a><span class="lineno"> 305</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00306" name="l00306"></a><span class="lineno"> 306</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00307" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00307" name="l00307"></a><span class="lineno"><a class="line" href="group__ops.html#gaf115782d009ac2a547fcca395c9ec797"> 307</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="namespacemlx_1_1core.html#a937503d72b66c661bf3f5fdcd98ef97c">operator==</a>(T a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b) {</div>
|
|
<div class="line"><a id="l00308" name="l00308"></a><span class="lineno"> 308</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga33638dc3a9972dd02be12d0eb85f9bde">equal</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(a), b);</div>
|
|
<div class="line"><a id="l00309" name="l00309"></a><span class="lineno"> 309</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00310" name="l00310"></a><span class="lineno"> 310</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00311" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00311" name="l00311"></a><span class="lineno"><a class="line" href="group__ops.html#ga3ad3ed7aece2650943a35082dbe3a0a5"> 311</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="namespacemlx_1_1core.html#a937503d72b66c661bf3f5fdcd98ef97c">operator==</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, T b) {</div>
|
|
<div class="line"><a id="l00312" name="l00312"></a><span class="lineno"> 312</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga33638dc3a9972dd02be12d0eb85f9bde">equal</a>(a, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(b));</div>
|
|
<div class="line"><a id="l00313" name="l00313"></a><span class="lineno"> 313</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00314" name="l00314"></a><span class="lineno"> 314</span> </div>
|
|
<div class="line"><a id="l00316" name="l00316"></a><span class="lineno"><a class="line" href="group__ops.html#ga28f22c5d2c399eee53be7b3facc11103"> 316</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga28f22c5d2c399eee53be7b3facc11103">not_equal</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00317" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00317" name="l00317"></a><span class="lineno"><a class="line" href="group__ops.html#ga0ac483d85f23252ca8757e9926d5a3c5"> 317</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="namespacemlx_1_1core.html#a94d00a1b7f8a4717ab3f26f45e4da655">operator!=</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b) {</div>
|
|
<div class="line"><a id="l00318" name="l00318"></a><span class="lineno"> 318</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga28f22c5d2c399eee53be7b3facc11103">not_equal</a>(a, b);</div>
|
|
<div class="line"><a id="l00319" name="l00319"></a><span class="lineno"> 319</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00320" name="l00320"></a><span class="lineno"> 320</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00321" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00321" name="l00321"></a><span class="lineno"><a class="line" href="group__ops.html#ga3fecba9f3cb9a19afd8ca492cf509ce0"> 321</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="namespacemlx_1_1core.html#a94d00a1b7f8a4717ab3f26f45e4da655">operator!=</a>(T a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b) {</div>
|
|
<div class="line"><a id="l00322" name="l00322"></a><span class="lineno"> 322</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga28f22c5d2c399eee53be7b3facc11103">not_equal</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(a), b);</div>
|
|
<div class="line"><a id="l00323" name="l00323"></a><span class="lineno"> 323</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00324" name="l00324"></a><span class="lineno"> 324</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00325" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00325" name="l00325"></a><span class="lineno"><a class="line" href="group__ops.html#gaebbf1cfde388c7480159a03c92c9a385"> 325</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="namespacemlx_1_1core.html#a94d00a1b7f8a4717ab3f26f45e4da655">operator!=</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, T b) {</div>
|
|
<div class="line"><a id="l00326" name="l00326"></a><span class="lineno"> 326</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga28f22c5d2c399eee53be7b3facc11103">not_equal</a>(a, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(b));</div>
|
|
<div class="line"><a id="l00327" name="l00327"></a><span class="lineno"> 327</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00328" name="l00328"></a><span class="lineno"> 328</span> </div>
|
|
<div class="line"><a id="l00330" name="l00330"></a><span class="lineno"><a class="line" href="group__ops.html#gaf4ec7bfc1ad13b891f1f3ef1772ef04d"> 330</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaf4ec7bfc1ad13b891f1f3ef1772ef04d">greater</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00331" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00331" name="l00331"></a><span class="lineno"><a class="line" href="group__ops.html#ga74fd2777adef10e6fe628a9cdadb01cb"> 331</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga74fd2777adef10e6fe628a9cdadb01cb">operator></a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b) {</div>
|
|
<div class="line"><a id="l00332" name="l00332"></a><span class="lineno"> 332</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gaf4ec7bfc1ad13b891f1f3ef1772ef04d">greater</a>(a, b);</div>
|
|
<div class="line"><a id="l00333" name="l00333"></a><span class="lineno"> 333</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00334" name="l00334"></a><span class="lineno"> 334</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00335" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00335" name="l00335"></a><span class="lineno"><a class="line" href="group__ops.html#ga32e106e794e2c32e4e7decee2df2477f"> 335</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga74fd2777adef10e6fe628a9cdadb01cb">operator></a>(T a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b) {</div>
|
|
<div class="line"><a id="l00336" name="l00336"></a><span class="lineno"> 336</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gaf4ec7bfc1ad13b891f1f3ef1772ef04d">greater</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(a), b);</div>
|
|
<div class="line"><a id="l00337" name="l00337"></a><span class="lineno"> 337</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00338" name="l00338"></a><span class="lineno"> 338</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00339" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00339" name="l00339"></a><span class="lineno"><a class="line" href="group__ops.html#ga96552b90e89923c5d2064cc427775ec5"> 339</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga74fd2777adef10e6fe628a9cdadb01cb">operator></a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, T b) {</div>
|
|
<div class="line"><a id="l00340" name="l00340"></a><span class="lineno"> 340</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gaf4ec7bfc1ad13b891f1f3ef1772ef04d">greater</a>(a, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(b));</div>
|
|
<div class="line"><a id="l00341" name="l00341"></a><span class="lineno"> 341</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00342" name="l00342"></a><span class="lineno"> 342</span> </div>
|
|
<div class="line"><a id="l00344" name="l00344"></a><span class="lineno"><a class="line" href="group__ops.html#ga7153071bcfff6faad21332163fb9a430"> 344</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7153071bcfff6faad21332163fb9a430">greater_equal</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00345" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00345" name="l00345"></a><span class="lineno"><a class="line" href="group__ops.html#ga3a41895f25ed083a36994d95fa102546"> 345</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga3a41895f25ed083a36994d95fa102546">operator>=</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b) {</div>
|
|
<div class="line"><a id="l00346" name="l00346"></a><span class="lineno"> 346</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga7153071bcfff6faad21332163fb9a430">greater_equal</a>(a, b);</div>
|
|
<div class="line"><a id="l00347" name="l00347"></a><span class="lineno"> 347</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00348" name="l00348"></a><span class="lineno"> 348</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00349" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00349" name="l00349"></a><span class="lineno"><a class="line" href="group__ops.html#gaf509f2cb3b18963232f20d6c3bd229b2"> 349</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga3a41895f25ed083a36994d95fa102546">operator>=</a>(T a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b) {</div>
|
|
<div class="line"><a id="l00350" name="l00350"></a><span class="lineno"> 350</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga7153071bcfff6faad21332163fb9a430">greater_equal</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(a), b);</div>
|
|
<div class="line"><a id="l00351" name="l00351"></a><span class="lineno"> 351</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00352" name="l00352"></a><span class="lineno"> 352</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00353" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00353" name="l00353"></a><span class="lineno"><a class="line" href="group__ops.html#gafa0eb25d5978674bfc9e59d4145ec590"> 353</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga3a41895f25ed083a36994d95fa102546">operator>=</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, T b) {</div>
|
|
<div class="line"><a id="l00354" name="l00354"></a><span class="lineno"> 354</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga7153071bcfff6faad21332163fb9a430">greater_equal</a>(a, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(b));</div>
|
|
<div class="line"><a id="l00355" name="l00355"></a><span class="lineno"> 355</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00356" name="l00356"></a><span class="lineno"> 356</span> </div>
|
|
<div class="line"><a id="l00358" name="l00358"></a><span class="lineno"><a class="line" href="group__ops.html#ga9142b8d717699a8abfa2a7398891ff8a"> 358</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga9142b8d717699a8abfa2a7398891ff8a">less</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00359" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00359" name="l00359"></a><span class="lineno"><a class="line" href="group__ops.html#gaee41e2b8f61d563200ff03575ac1d6c3"> 359</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaee41e2b8f61d563200ff03575ac1d6c3">operator<</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b) {</div>
|
|
<div class="line"><a id="l00360" name="l00360"></a><span class="lineno"> 360</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga9142b8d717699a8abfa2a7398891ff8a">less</a>(a, b);</div>
|
|
<div class="line"><a id="l00361" name="l00361"></a><span class="lineno"> 361</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00362" name="l00362"></a><span class="lineno"> 362</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00363" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00363" name="l00363"></a><span class="lineno"><a class="line" href="group__ops.html#ga1ef8ea11cf15ce628c54201fa42748ef"> 363</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaee41e2b8f61d563200ff03575ac1d6c3">operator<</a>(T a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b) {</div>
|
|
<div class="line"><a id="l00364" name="l00364"></a><span class="lineno"> 364</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga9142b8d717699a8abfa2a7398891ff8a">less</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(a), b);</div>
|
|
<div class="line"><a id="l00365" name="l00365"></a><span class="lineno"> 365</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00366" name="l00366"></a><span class="lineno"> 366</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00367" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00367" name="l00367"></a><span class="lineno"><a class="line" href="group__ops.html#ga95e72226dc7a79c40b3d16f990922050"> 367</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaee41e2b8f61d563200ff03575ac1d6c3">operator<</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, T b) {</div>
|
|
<div class="line"><a id="l00368" name="l00368"></a><span class="lineno"> 368</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga9142b8d717699a8abfa2a7398891ff8a">less</a>(a, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(b));</div>
|
|
<div class="line"><a id="l00369" name="l00369"></a><span class="lineno"> 369</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00370" name="l00370"></a><span class="lineno"> 370</span> </div>
|
|
<div class="line"><a id="l00372" name="l00372"></a><span class="lineno"><a class="line" href="group__ops.html#ga0d49e0c7011d0573c369c13c8f045a09"> 372</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga0d49e0c7011d0573c369c13c8f045a09">less_equal</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00373" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00373" name="l00373"></a><span class="lineno"><a class="line" href="group__ops.html#ga4c8b8a1632944acaae50f0de6c23ece6"> 373</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga4c8b8a1632944acaae50f0de6c23ece6">operator<=</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b) {</div>
|
|
<div class="line"><a id="l00374" name="l00374"></a><span class="lineno"> 374</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga0d49e0c7011d0573c369c13c8f045a09">less_equal</a>(a, b);</div>
|
|
<div class="line"><a id="l00375" name="l00375"></a><span class="lineno"> 375</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00376" name="l00376"></a><span class="lineno"> 376</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00377" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00377" name="l00377"></a><span class="lineno"><a class="line" href="group__ops.html#ga150a9be467c9f91482a6d6fc13504bc4"> 377</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga4c8b8a1632944acaae50f0de6c23ece6">operator<=</a>(T a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b) {</div>
|
|
<div class="line"><a id="l00378" name="l00378"></a><span class="lineno"> 378</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga0d49e0c7011d0573c369c13c8f045a09">less_equal</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(a), b);</div>
|
|
<div class="line"><a id="l00379" name="l00379"></a><span class="lineno"> 379</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00380" name="l00380"></a><span class="lineno"> 380</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00381" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00381" name="l00381"></a><span class="lineno"><a class="line" href="group__ops.html#ga624eeccef0cc4b130e1325abfea057cb"> 381</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga4c8b8a1632944acaae50f0de6c23ece6">operator<=</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, T b) {</div>
|
|
<div class="line"><a id="l00382" name="l00382"></a><span class="lineno"> 382</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga0d49e0c7011d0573c369c13c8f045a09">less_equal</a>(a, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(b));</div>
|
|
<div class="line"><a id="l00383" name="l00383"></a><span class="lineno"> 383</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00384" name="l00384"></a><span class="lineno"> 384</span> </div>
|
|
<div class="line"><a id="l00386" name="l00386"></a><span class="lineno"><a class="line" href="group__ops.html#ga8f3059336ee0c87207b1f8c6ab312645"> 386</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga8f3059336ee0c87207b1f8c6ab312645">array_equal</a>(</div>
|
|
<div class="line"><a id="l00387" name="l00387"></a><span class="lineno"> 387</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00388" name="l00388"></a><span class="lineno"> 388</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b,</div>
|
|
<div class="line"><a id="l00389" name="l00389"></a><span class="lineno"> 389</span> <span class="keywordtype">bool</span> equal_nan,</div>
|
|
<div class="line"><a id="l00390" name="l00390"></a><span class="lineno"> 390</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00391" name="l00391"></a><span class="lineno"> 391</span><span class="keyword">inline</span> array</div>
|
|
<div class="foldopen" id="foldopen00392" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00392" name="l00392"></a><span class="lineno"><a class="line" href="group__ops.html#gaf79cf0271ca0105d7b14295a90d0ed14"> 392</a></span><a class="code hl_function" href="group__ops.html#ga8f3059336ee0c87207b1f8c6ab312645">array_equal</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00393" name="l00393"></a><span class="lineno"> 393</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga8f3059336ee0c87207b1f8c6ab312645">array_equal</a>(a, b, <span class="keyword">false</span>, s);</div>
|
|
<div class="line"><a id="l00394" name="l00394"></a><span class="lineno"> 394</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00395" name="l00395"></a><span class="lineno"> 395</span> </div>
|
|
<div class="line"><a id="l00396" name="l00396"></a><span class="lineno"><a class="line" href="group__ops.html#ga175592792471b0ffb45196dca4711ba6"> 396</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga175592792471b0ffb45196dca4711ba6">isnan</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00397" name="l00397"></a><span class="lineno"> 397</span> </div>
|
|
<div class="line"><a id="l00398" name="l00398"></a><span class="lineno"><a class="line" href="group__ops.html#ga8fc238d5e5d1153e69da8b36015d9844"> 398</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga8fc238d5e5d1153e69da8b36015d9844">isinf</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00399" name="l00399"></a><span class="lineno"> 399</span> </div>
|
|
<div class="line"><a id="l00400" name="l00400"></a><span class="lineno"><a class="line" href="group__ops.html#ga725ff0789f934b1fdd54ee29e47022ff"> 400</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga725ff0789f934b1fdd54ee29e47022ff">isfinite</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00401" name="l00401"></a><span class="lineno"> 401</span> </div>
|
|
<div class="line"><a id="l00402" name="l00402"></a><span class="lineno"><a class="line" href="group__ops.html#gad80f7c4a58c12b6cb30a8b9a73008993"> 402</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gad80f7c4a58c12b6cb30a8b9a73008993">isposinf</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00403" name="l00403"></a><span class="lineno"> 403</span> </div>
|
|
<div class="line"><a id="l00404" name="l00404"></a><span class="lineno"><a class="line" href="group__ops.html#ga1940523da381ed7be50656a3bc465ff3"> 404</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga1940523da381ed7be50656a3bc465ff3">isneginf</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00405" name="l00405"></a><span class="lineno"> 405</span> </div>
|
|
<div class="line"><a id="l00407" name="l00407"></a><span class="lineno"><a class="line" href="group__ops.html#ga8a2056f8c9bb30914c40bcf509386491"> 407</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga8a2056f8c9bb30914c40bcf509386491">where</a>(</div>
|
|
<div class="line"><a id="l00408" name="l00408"></a><span class="lineno"> 408</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& condition,</div>
|
|
<div class="line"><a id="l00409" name="l00409"></a><span class="lineno"> 409</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& x,</div>
|
|
<div class="line"><a id="l00410" name="l00410"></a><span class="lineno"> 410</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& y,</div>
|
|
<div class="line"><a id="l00411" name="l00411"></a><span class="lineno"> 411</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00412" name="l00412"></a><span class="lineno"> 412</span> </div>
|
|
<div class="line"><a id="l00414" name="l00414"></a><span class="lineno"><a class="line" href="group__ops.html#gab1467c6a9e675152e768afd6dcfb61de"> 414</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gab1467c6a9e675152e768afd6dcfb61de">nan_to_num</a>(</div>
|
|
<div class="line"><a id="l00415" name="l00415"></a><span class="lineno"> 415</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00416" name="l00416"></a><span class="lineno"> 416</span> <span class="keywordtype">float</span> nan = 0.0f,</div>
|
|
<div class="line"><a id="l00417" name="l00417"></a><span class="lineno"> 417</span> <span class="keyword">const</span> std::optional<float> posinf = std::nullopt,</div>
|
|
<div class="line"><a id="l00418" name="l00418"></a><span class="lineno"> 418</span> <span class="keyword">const</span> std::optional<float> neginf = std::nullopt,</div>
|
|
<div class="line"><a id="l00419" name="l00419"></a><span class="lineno"> 419</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00420" name="l00420"></a><span class="lineno"> 420</span> </div>
|
|
<div class="line"><a id="l00422" name="l00422"></a><span class="lineno"><a class="line" href="group__ops.html#ga3b1b90ef1275ca17655b6d7f25d3ee68"> 422</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga3b1b90ef1275ca17655b6d7f25d3ee68">all</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">bool</span> keepdims, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00423" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00423" name="l00423"></a><span class="lineno"><a class="line" href="group__ops.html#ga3689e12e8f42dadb4cbe2b07dc4099f4"> 423</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga3b1b90ef1275ca17655b6d7f25d3ee68">all</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00424" name="l00424"></a><span class="lineno"> 424</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga3b1b90ef1275ca17655b6d7f25d3ee68">all</a>(a, <span class="keyword">false</span>, <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00425" name="l00425"></a><span class="lineno"> 425</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00426" name="l00426"></a><span class="lineno"> 426</span> </div>
|
|
<div class="line"><a id="l00428" name="l00428"></a><span class="lineno"><a class="line" href="group__ops.html#gaf0cd4257de7542daf9faf5e605e31020"> 428</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaf0cd4257de7542daf9faf5e605e31020">allclose</a>(</div>
|
|
<div class="line"><a id="l00429" name="l00429"></a><span class="lineno"> 429</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00430" name="l00430"></a><span class="lineno"> 430</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b,</div>
|
|
<div class="line"><a id="l00431" name="l00431"></a><span class="lineno"> 431</span> <span class="keywordtype">double</span> rtol = 1e-5,</div>
|
|
<div class="line"><a id="l00432" name="l00432"></a><span class="lineno"> 432</span> <span class="keywordtype">double</span> atol = 1e-8,</div>
|
|
<div class="line"><a id="l00433" name="l00433"></a><span class="lineno"> 433</span> <span class="keywordtype">bool</span> equal_nan = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00434" name="l00434"></a><span class="lineno"> 434</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00435" name="l00435"></a><span class="lineno"> 435</span> </div>
|
|
<div class="line"><a id="l00438" name="l00438"></a><span class="lineno"><a class="line" href="group__ops.html#ga51eac95c04400921c54716de14b52491"> 438</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga51eac95c04400921c54716de14b52491">isclose</a>(</div>
|
|
<div class="line"><a id="l00439" name="l00439"></a><span class="lineno"> 439</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00440" name="l00440"></a><span class="lineno"> 440</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b,</div>
|
|
<div class="line"><a id="l00441" name="l00441"></a><span class="lineno"> 441</span> <span class="keywordtype">double</span> rtol = 1e-5,</div>
|
|
<div class="line"><a id="l00442" name="l00442"></a><span class="lineno"> 442</span> <span class="keywordtype">double</span> atol = 1e-8,</div>
|
|
<div class="line"><a id="l00443" name="l00443"></a><span class="lineno"> 443</span> <span class="keywordtype">bool</span> equal_nan = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00444" name="l00444"></a><span class="lineno"> 444</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00445" name="l00445"></a><span class="lineno"> 445</span> </div>
|
|
<div class="line"><a id="l00450" name="l00450"></a><span class="lineno"><a class="line" href="group__ops.html#gac0919c6ba53aea35a7683dea7e9a9a59"> 450</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga3b1b90ef1275ca17655b6d7f25d3ee68">all</a>(</div>
|
|
<div class="line"><a id="l00451" name="l00451"></a><span class="lineno"> 451</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00452" name="l00452"></a><span class="lineno"> 452</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00453" name="l00453"></a><span class="lineno"> 453</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00454" name="l00454"></a><span class="lineno"> 454</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00455" name="l00455"></a><span class="lineno"> 455</span> </div>
|
|
<div class="line"><a id="l00460" name="l00460"></a><span class="lineno"><a class="line" href="group__ops.html#gae2d5fcc5b62d673cca76c08b7b4afbbc"> 460</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga3b1b90ef1275ca17655b6d7f25d3ee68">all</a>(</div>
|
|
<div class="line"><a id="l00461" name="l00461"></a><span class="lineno"> 461</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00462" name="l00462"></a><span class="lineno"> 462</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00463" name="l00463"></a><span class="lineno"> 463</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00464" name="l00464"></a><span class="lineno"> 464</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00465" name="l00465"></a><span class="lineno"> 465</span> </div>
|
|
<div class="line"><a id="l00467" name="l00467"></a><span class="lineno"><a class="line" href="group__ops.html#ga8598dd718fb05cb28535e250372d4e6f"> 467</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga8598dd718fb05cb28535e250372d4e6f">any</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">bool</span> keepdims, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00468" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00468" name="l00468"></a><span class="lineno"><a class="line" href="group__ops.html#gad37df97f253a963bece124198dbaf9ba"> 468</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga8598dd718fb05cb28535e250372d4e6f">any</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00469" name="l00469"></a><span class="lineno"> 469</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga8598dd718fb05cb28535e250372d4e6f">any</a>(a, <span class="keyword">false</span>, <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00470" name="l00470"></a><span class="lineno"> 470</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00471" name="l00471"></a><span class="lineno"> 471</span> </div>
|
|
<div class="line"><a id="l00476" name="l00476"></a><span class="lineno"><a class="line" href="group__ops.html#gaf240618fc8b06debf5f56e97e84f18ef"> 476</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga8598dd718fb05cb28535e250372d4e6f">any</a>(</div>
|
|
<div class="line"><a id="l00477" name="l00477"></a><span class="lineno"> 477</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00478" name="l00478"></a><span class="lineno"> 478</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00479" name="l00479"></a><span class="lineno"> 479</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00480" name="l00480"></a><span class="lineno"> 480</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00481" name="l00481"></a><span class="lineno"> 481</span> </div>
|
|
<div class="line"><a id="l00486" name="l00486"></a><span class="lineno"><a class="line" href="group__ops.html#gab1d56277d468a55227f4dad6bc2fc1ce"> 486</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga8598dd718fb05cb28535e250372d4e6f">any</a>(</div>
|
|
<div class="line"><a id="l00487" name="l00487"></a><span class="lineno"> 487</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00488" name="l00488"></a><span class="lineno"> 488</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00489" name="l00489"></a><span class="lineno"> 489</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00490" name="l00490"></a><span class="lineno"> 490</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00491" name="l00491"></a><span class="lineno"> 491</span> </div>
|
|
<div class="line"><a id="l00493" name="l00493"></a><span class="lineno"><a class="line" href="group__ops.html#gade905ee92eb6ab7edfc312aeddfbaeb6"> 493</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gade905ee92eb6ab7edfc312aeddfbaeb6">sum</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">bool</span> keepdims, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00494" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00494" name="l00494"></a><span class="lineno"><a class="line" href="group__ops.html#ga3627754d7868487bdab1bd83f05d9c81"> 494</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gade905ee92eb6ab7edfc312aeddfbaeb6">sum</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00495" name="l00495"></a><span class="lineno"> 495</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gade905ee92eb6ab7edfc312aeddfbaeb6">sum</a>(a, <span class="keyword">false</span>, <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00496" name="l00496"></a><span class="lineno"> 496</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00497" name="l00497"></a><span class="lineno"> 497</span> </div>
|
|
<div class="line"><a id="l00499" name="l00499"></a><span class="lineno"><a class="line" href="group__ops.html#gaccd0a6be2c5b5128fdc2d87b5c8e67f4"> 499</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gade905ee92eb6ab7edfc312aeddfbaeb6">sum</a>(</div>
|
|
<div class="line"><a id="l00500" name="l00500"></a><span class="lineno"> 500</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00501" name="l00501"></a><span class="lineno"> 501</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00502" name="l00502"></a><span class="lineno"> 502</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00503" name="l00503"></a><span class="lineno"> 503</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00504" name="l00504"></a><span class="lineno"> 504</span> </div>
|
|
<div class="line"><a id="l00506" name="l00506"></a><span class="lineno"><a class="line" href="group__ops.html#gafcd39b0bf39a56c26a967981c7ab8a8d"> 506</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gade905ee92eb6ab7edfc312aeddfbaeb6">sum</a>(</div>
|
|
<div class="line"><a id="l00507" name="l00507"></a><span class="lineno"> 507</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00508" name="l00508"></a><span class="lineno"> 508</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00509" name="l00509"></a><span class="lineno"> 509</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00510" name="l00510"></a><span class="lineno"> 510</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00511" name="l00511"></a><span class="lineno"> 511</span> </div>
|
|
<div class="line"><a id="l00513" name="l00513"></a><span class="lineno"><a class="line" href="group__ops.html#gade46e768fd46b8b640eb16f26abeecef"> 513</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gade46e768fd46b8b640eb16f26abeecef">mean</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">bool</span> keepdims, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00514" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00514" name="l00514"></a><span class="lineno"><a class="line" href="group__ops.html#ga52b59fdd8e8430538e564f5bbcfa31e6"> 514</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gade46e768fd46b8b640eb16f26abeecef">mean</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00515" name="l00515"></a><span class="lineno"> 515</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gade46e768fd46b8b640eb16f26abeecef">mean</a>(a, <span class="keyword">false</span>, <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00516" name="l00516"></a><span class="lineno"> 516</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00517" name="l00517"></a><span class="lineno"> 517</span> </div>
|
|
<div class="line"><a id="l00519" name="l00519"></a><span class="lineno"><a class="line" href="group__ops.html#ga066161f3d3e395a1d76c638cb680d444"> 519</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gade46e768fd46b8b640eb16f26abeecef">mean</a>(</div>
|
|
<div class="line"><a id="l00520" name="l00520"></a><span class="lineno"> 520</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00521" name="l00521"></a><span class="lineno"> 521</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00522" name="l00522"></a><span class="lineno"> 522</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00523" name="l00523"></a><span class="lineno"> 523</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00524" name="l00524"></a><span class="lineno"> 524</span> </div>
|
|
<div class="line"><a id="l00526" name="l00526"></a><span class="lineno"><a class="line" href="group__ops.html#ga45fba73eab0e3b6e128ed3ce2f43a5da"> 526</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gade46e768fd46b8b640eb16f26abeecef">mean</a>(</div>
|
|
<div class="line"><a id="l00527" name="l00527"></a><span class="lineno"> 527</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00528" name="l00528"></a><span class="lineno"> 528</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00529" name="l00529"></a><span class="lineno"> 529</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00530" name="l00530"></a><span class="lineno"> 530</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00531" name="l00531"></a><span class="lineno"> 531</span> </div>
|
|
<div class="line"><a id="l00533" name="l00533"></a><span class="lineno"><a class="line" href="group__ops.html#ga7e133df686439588a8cd1fb10ce0c6e9"> 533</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7e133df686439588a8cd1fb10ce0c6e9">var</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">bool</span> keepdims, <span class="keywordtype">int</span> ddof = 0, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00534" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00534" name="l00534"></a><span class="lineno"><a class="line" href="group__ops.html#ga7d7b38d118fa2613214078ef0f7d5a42"> 534</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7e133df686439588a8cd1fb10ce0c6e9">var</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00535" name="l00535"></a><span class="lineno"> 535</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga7e133df686439588a8cd1fb10ce0c6e9">var</a>(a, <span class="keyword">false</span>, 0, <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00536" name="l00536"></a><span class="lineno"> 536</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00537" name="l00537"></a><span class="lineno"> 537</span> </div>
|
|
<div class="line"><a id="l00540" name="l00540"></a><span class="lineno"><a class="line" href="group__ops.html#ga78ddeb966cbe7a5b0aa17e1de43025f2"> 540</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7e133df686439588a8cd1fb10ce0c6e9">var</a>(</div>
|
|
<div class="line"><a id="l00541" name="l00541"></a><span class="lineno"> 541</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00542" name="l00542"></a><span class="lineno"> 542</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00543" name="l00543"></a><span class="lineno"> 543</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00544" name="l00544"></a><span class="lineno"> 544</span> <span class="keywordtype">int</span> ddof = 0,</div>
|
|
<div class="line"><a id="l00545" name="l00545"></a><span class="lineno"> 545</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00546" name="l00546"></a><span class="lineno"> 546</span> </div>
|
|
<div class="line"><a id="l00549" name="l00549"></a><span class="lineno"><a class="line" href="group__ops.html#ga4fbf3e3f98f2e4956faf87af320aa9d0"> 549</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7e133df686439588a8cd1fb10ce0c6e9">var</a>(</div>
|
|
<div class="line"><a id="l00550" name="l00550"></a><span class="lineno"> 550</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00551" name="l00551"></a><span class="lineno"> 551</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00552" name="l00552"></a><span class="lineno"> 552</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00553" name="l00553"></a><span class="lineno"> 553</span> <span class="keywordtype">int</span> ddof = 0,</div>
|
|
<div class="line"><a id="l00554" name="l00554"></a><span class="lineno"> 554</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00555" name="l00555"></a><span class="lineno"> 555</span> </div>
|
|
<div class="line"><a id="l00557" name="l00557"></a><span class="lineno"><a class="line" href="group__ops.html#ga2a466024f8061febc0a64be557644cb0"> 557</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga2a466024f8061febc0a64be557644cb0">std</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">bool</span> keepdims, <span class="keywordtype">int</span> ddof = 0, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00558" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00558" name="l00558"></a><span class="lineno"><a class="line" href="group__ops.html#gafdcb04d77c64405a3990078a77dd984c"> 558</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga2a466024f8061febc0a64be557644cb0">std</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00559" name="l00559"></a><span class="lineno"> 559</span> <span class="keywordflow">return</span> std(a, <span class="keyword">false</span>, 0, <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00560" name="l00560"></a><span class="lineno"> 560</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00561" name="l00561"></a><span class="lineno"> 561</span> </div>
|
|
<div class="line"><a id="l00564" name="l00564"></a><span class="lineno"><a class="line" href="group__ops.html#ga7f649970bf38b987b6ef847054f3c2f8"> 564</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga2a466024f8061febc0a64be557644cb0">std</a>(</div>
|
|
<div class="line"><a id="l00565" name="l00565"></a><span class="lineno"> 565</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00566" name="l00566"></a><span class="lineno"> 566</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00567" name="l00567"></a><span class="lineno"> 567</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00568" name="l00568"></a><span class="lineno"> 568</span> <span class="keywordtype">int</span> ddof = 0,</div>
|
|
<div class="line"><a id="l00569" name="l00569"></a><span class="lineno"> 569</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00570" name="l00570"></a><span class="lineno"> 570</span> </div>
|
|
<div class="line"><a id="l00573" name="l00573"></a><span class="lineno"><a class="line" href="group__ops.html#ga62721a206df671ef5797449eea97af9f"> 573</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga2a466024f8061febc0a64be557644cb0">std</a>(</div>
|
|
<div class="line"><a id="l00574" name="l00574"></a><span class="lineno"> 574</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00575" name="l00575"></a><span class="lineno"> 575</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00576" name="l00576"></a><span class="lineno"> 576</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00577" name="l00577"></a><span class="lineno"> 577</span> <span class="keywordtype">int</span> ddof = 0,</div>
|
|
<div class="line"><a id="l00578" name="l00578"></a><span class="lineno"> 578</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00579" name="l00579"></a><span class="lineno"> 579</span> </div>
|
|
<div class="line"><a id="l00581" name="l00581"></a><span class="lineno"><a class="line" href="group__ops.html#ga4a09b7241d564d92548bc2773eb1d544"> 581</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga4a09b7241d564d92548bc2773eb1d544">prod</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">bool</span> keepdims, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00582" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00582" name="l00582"></a><span class="lineno"><a class="line" href="group__ops.html#ga61832191f3c42ea549cf04953edc3602"> 582</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga4a09b7241d564d92548bc2773eb1d544">prod</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00583" name="l00583"></a><span class="lineno"> 583</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga4a09b7241d564d92548bc2773eb1d544">prod</a>(a, <span class="keyword">false</span>, <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00584" name="l00584"></a><span class="lineno"> 584</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00585" name="l00585"></a><span class="lineno"> 585</span> </div>
|
|
<div class="line"><a id="l00587" name="l00587"></a><span class="lineno"><a class="line" href="group__ops.html#ga2b3935108f641e20a70dbf63f540d970"> 587</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga4a09b7241d564d92548bc2773eb1d544">prod</a>(</div>
|
|
<div class="line"><a id="l00588" name="l00588"></a><span class="lineno"> 588</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00589" name="l00589"></a><span class="lineno"> 589</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00590" name="l00590"></a><span class="lineno"> 590</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00591" name="l00591"></a><span class="lineno"> 591</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00592" name="l00592"></a><span class="lineno"> 592</span> </div>
|
|
<div class="line"><a id="l00594" name="l00594"></a><span class="lineno"><a class="line" href="group__ops.html#ga8a10a10b81c69996d0aca8ba401f8ff0"> 594</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga4a09b7241d564d92548bc2773eb1d544">prod</a>(</div>
|
|
<div class="line"><a id="l00595" name="l00595"></a><span class="lineno"> 595</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00596" name="l00596"></a><span class="lineno"> 596</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00597" name="l00597"></a><span class="lineno"> 597</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00598" name="l00598"></a><span class="lineno"> 598</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00599" name="l00599"></a><span class="lineno"> 599</span> </div>
|
|
<div class="line"><a id="l00601" name="l00601"></a><span class="lineno"><a class="line" href="group__ops.html#ga7fed87d96cc7741d8267f4eac83f5fe7"> 601</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7fed87d96cc7741d8267f4eac83f5fe7">max</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">bool</span> keepdims, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00602" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00602" name="l00602"></a><span class="lineno"><a class="line" href="group__ops.html#ga25be91d70a5f40341db0615a0b8bfedc"> 602</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7fed87d96cc7741d8267f4eac83f5fe7">max</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00603" name="l00603"></a><span class="lineno"> 603</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga7fed87d96cc7741d8267f4eac83f5fe7">max</a>(a, <span class="keyword">false</span>, <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00604" name="l00604"></a><span class="lineno"> 604</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00605" name="l00605"></a><span class="lineno"> 605</span> </div>
|
|
<div class="line"><a id="l00607" name="l00607"></a><span class="lineno"><a class="line" href="group__ops.html#ga1ca7b6b91fe2459a7d83897bf013827f"> 607</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7fed87d96cc7741d8267f4eac83f5fe7">max</a>(</div>
|
|
<div class="line"><a id="l00608" name="l00608"></a><span class="lineno"> 608</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00609" name="l00609"></a><span class="lineno"> 609</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00610" name="l00610"></a><span class="lineno"> 610</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00611" name="l00611"></a><span class="lineno"> 611</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00612" name="l00612"></a><span class="lineno"> 612</span> </div>
|
|
<div class="line"><a id="l00614" name="l00614"></a><span class="lineno"><a class="line" href="group__ops.html#ga7b638050e03a93f2896c981bc2850a47"> 614</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7fed87d96cc7741d8267f4eac83f5fe7">max</a>(</div>
|
|
<div class="line"><a id="l00615" name="l00615"></a><span class="lineno"> 615</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00616" name="l00616"></a><span class="lineno"> 616</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00617" name="l00617"></a><span class="lineno"> 617</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00618" name="l00618"></a><span class="lineno"> 618</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00619" name="l00619"></a><span class="lineno"> 619</span> </div>
|
|
<div class="line"><a id="l00621" name="l00621"></a><span class="lineno"><a class="line" href="group__ops.html#gab27599802617a4c8f9964ab5f4ffee12"> 621</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gab27599802617a4c8f9964ab5f4ffee12">min</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">bool</span> keepdims, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00622" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00622" name="l00622"></a><span class="lineno"><a class="line" href="group__ops.html#ga0140b91e9cdfc3fef0da8e332f65a9e8"> 622</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gab27599802617a4c8f9964ab5f4ffee12">min</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00623" name="l00623"></a><span class="lineno"> 623</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gab27599802617a4c8f9964ab5f4ffee12">min</a>(a, <span class="keyword">false</span>, <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00624" name="l00624"></a><span class="lineno"> 624</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00625" name="l00625"></a><span class="lineno"> 625</span> </div>
|
|
<div class="line"><a id="l00627" name="l00627"></a><span class="lineno"><a class="line" href="group__ops.html#ga6efb83cd46436678c8f8c4af15cc00f5"> 627</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gab27599802617a4c8f9964ab5f4ffee12">min</a>(</div>
|
|
<div class="line"><a id="l00628" name="l00628"></a><span class="lineno"> 628</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00629" name="l00629"></a><span class="lineno"> 629</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00630" name="l00630"></a><span class="lineno"> 630</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00631" name="l00631"></a><span class="lineno"> 631</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00632" name="l00632"></a><span class="lineno"> 632</span> </div>
|
|
<div class="line"><a id="l00634" name="l00634"></a><span class="lineno"><a class="line" href="group__ops.html#ga36fa315eef677f4143868f552cd26d03"> 634</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gab27599802617a4c8f9964ab5f4ffee12">min</a>(</div>
|
|
<div class="line"><a id="l00635" name="l00635"></a><span class="lineno"> 635</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00636" name="l00636"></a><span class="lineno"> 636</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00637" name="l00637"></a><span class="lineno"> 637</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00638" name="l00638"></a><span class="lineno"> 638</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00639" name="l00639"></a><span class="lineno"> 639</span> </div>
|
|
<div class="line"><a id="l00641" name="l00641"></a><span class="lineno"><a class="line" href="group__ops.html#ga7c3bd5ef430a71dfd298e626741e3c71"> 641</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7c3bd5ef430a71dfd298e626741e3c71">argmin</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">bool</span> keepdims, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00642" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00642" name="l00642"></a><span class="lineno"><a class="line" href="group__ops.html#ga6bc577c5ab10cd9c848ba81321595070"> 642</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7c3bd5ef430a71dfd298e626741e3c71">argmin</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00643" name="l00643"></a><span class="lineno"> 643</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga7c3bd5ef430a71dfd298e626741e3c71">argmin</a>(a, <span class="keyword">false</span>, s);</div>
|
|
<div class="line"><a id="l00644" name="l00644"></a><span class="lineno"> 644</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00645" name="l00645"></a><span class="lineno"> 645</span> </div>
|
|
<div class="line"><a id="l00647" name="l00647"></a><span class="lineno"><a class="line" href="group__ops.html#gaf66dc3c77b88e4009e0678eda41eca81"> 647</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7c3bd5ef430a71dfd298e626741e3c71">argmin</a>(</div>
|
|
<div class="line"><a id="l00648" name="l00648"></a><span class="lineno"> 648</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00649" name="l00649"></a><span class="lineno"> 649</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00650" name="l00650"></a><span class="lineno"> 650</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00651" name="l00651"></a><span class="lineno"> 651</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00652" name="l00652"></a><span class="lineno"> 652</span> </div>
|
|
<div class="line"><a id="l00654" name="l00654"></a><span class="lineno"><a class="line" href="group__ops.html#gae60b0b5339b9c50b9970260faf613e83"> 654</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gae60b0b5339b9c50b9970260faf613e83">argmax</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">bool</span> keepdims, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00655" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00655" name="l00655"></a><span class="lineno"><a class="line" href="group__ops.html#gae6f6c5a840320b336fdc9687e0ed56c8"> 655</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gae60b0b5339b9c50b9970260faf613e83">argmax</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00656" name="l00656"></a><span class="lineno"> 656</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gae60b0b5339b9c50b9970260faf613e83">argmax</a>(a, <span class="keyword">false</span>, s);</div>
|
|
<div class="line"><a id="l00657" name="l00657"></a><span class="lineno"> 657</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00658" name="l00658"></a><span class="lineno"> 658</span> </div>
|
|
<div class="line"><a id="l00660" name="l00660"></a><span class="lineno"><a class="line" href="group__ops.html#ga2efa67466510fc26ab9ea8dff30f2ba5"> 660</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gae60b0b5339b9c50b9970260faf613e83">argmax</a>(</div>
|
|
<div class="line"><a id="l00661" name="l00661"></a><span class="lineno"> 661</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00662" name="l00662"></a><span class="lineno"> 662</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00663" name="l00663"></a><span class="lineno"> 663</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00664" name="l00664"></a><span class="lineno"> 664</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00665" name="l00665"></a><span class="lineno"> 665</span> </div>
|
|
<div class="line"><a id="l00667" name="l00667"></a><span class="lineno"><a class="line" href="group__ops.html#ga7fb616054665b3c2d61fa234f501f079"> 667</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7fb616054665b3c2d61fa234f501f079">sort</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00668" name="l00668"></a><span class="lineno"> 668</span> </div>
|
|
<div class="line"><a id="l00670" name="l00670"></a><span class="lineno"><a class="line" href="group__ops.html#gaae1bc47aa737f705d0e5884270063fea"> 670</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7fb616054665b3c2d61fa234f501f079">sort</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> axis, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00671" name="l00671"></a><span class="lineno"> 671</span> </div>
|
|
<div class="line"><a id="l00673" name="l00673"></a><span class="lineno"><a class="line" href="group__ops.html#ga8df3b2703bf671457422894dd870cdc5"> 673</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga8df3b2703bf671457422894dd870cdc5">argsort</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00674" name="l00674"></a><span class="lineno"> 674</span> </div>
|
|
<div class="line"><a id="l00676" name="l00676"></a><span class="lineno"><a class="line" href="group__ops.html#ga7878e0daa5a75f44e57b5fe948fa3ef6"> 676</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga8df3b2703bf671457422894dd870cdc5">argsort</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> axis, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00677" name="l00677"></a><span class="lineno"> 677</span> </div>
|
|
<div class="line"><a id="l00682" name="l00682"></a><span class="lineno"><a class="line" href="group__ops.html#gac1b30830a972fb9a2601379ad2b32405"> 682</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac1b30830a972fb9a2601379ad2b32405">partition</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> kth, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00683" name="l00683"></a><span class="lineno"> 683</span> </div>
|
|
<div class="line"><a id="l00688" name="l00688"></a><span class="lineno"><a class="line" href="group__ops.html#ga4fbea3a5f66cf81e3c119d1661119321"> 688</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac1b30830a972fb9a2601379ad2b32405">partition</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> kth, <span class="keywordtype">int</span> axis, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00689" name="l00689"></a><span class="lineno"> 689</span> </div>
|
|
<div class="line"><a id="l00694" name="l00694"></a><span class="lineno"><a class="line" href="group__ops.html#gaf301c49c10fa9b95a9e8dc52ead1a8dd"> 694</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaf301c49c10fa9b95a9e8dc52ead1a8dd">argpartition</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> kth, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00695" name="l00695"></a><span class="lineno"> 695</span> </div>
|
|
<div class="line"><a id="l00700" name="l00700"></a><span class="lineno"><a class="line" href="group__ops.html#ga7b15c654c7463def57857a0e239989a3"> 700</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaf301c49c10fa9b95a9e8dc52ead1a8dd">argpartition</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> kth, <span class="keywordtype">int</span> axis, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00701" name="l00701"></a><span class="lineno"> 701</span> </div>
|
|
<div class="line"><a id="l00703" name="l00703"></a><span class="lineno"><a class="line" href="group__ops.html#ga5487dd887c43e5341f3e68ffe47f0f5a"> 703</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga5487dd887c43e5341f3e68ffe47f0f5a">topk</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> k, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00704" name="l00704"></a><span class="lineno"> 704</span> </div>
|
|
<div class="line"><a id="l00706" name="l00706"></a><span class="lineno"><a class="line" href="group__ops.html#ga35b8436c79ff953f6c809598b646f498"> 706</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga5487dd887c43e5341f3e68ffe47f0f5a">topk</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> k, <span class="keywordtype">int</span> axis, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00707" name="l00707"></a><span class="lineno"> 707</span> </div>
|
|
<div class="line"><a id="l00709" name="l00709"></a><span class="lineno"><a class="line" href="group__ops.html#gacff4eb57c085d571e722083680267ac5"> 709</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gacff4eb57c085d571e722083680267ac5">logsumexp</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">bool</span> keepdims, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00710" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00710" name="l00710"></a><span class="lineno"><a class="line" href="group__ops.html#ga59be50b4e92f1dc20b53460cefa3910d"> 710</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gacff4eb57c085d571e722083680267ac5">logsumexp</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00711" name="l00711"></a><span class="lineno"> 711</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gacff4eb57c085d571e722083680267ac5">logsumexp</a>(a, <span class="keyword">false</span>, <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00712" name="l00712"></a><span class="lineno"> 712</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00713" name="l00713"></a><span class="lineno"> 713</span> </div>
|
|
<div class="line"><a id="l00715" name="l00715"></a><span class="lineno"><a class="line" href="group__ops.html#gae3969c7bd24c4f3ab97831df28239689"> 715</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gacff4eb57c085d571e722083680267ac5">logsumexp</a>(</div>
|
|
<div class="line"><a id="l00716" name="l00716"></a><span class="lineno"> 716</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00717" name="l00717"></a><span class="lineno"> 717</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00718" name="l00718"></a><span class="lineno"> 718</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00719" name="l00719"></a><span class="lineno"> 719</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00720" name="l00720"></a><span class="lineno"> 720</span> </div>
|
|
<div class="line"><a id="l00722" name="l00722"></a><span class="lineno"><a class="line" href="group__ops.html#gafef5cb2159c16a60a95470cc823bdd44"> 722</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gacff4eb57c085d571e722083680267ac5">logsumexp</a>(</div>
|
|
<div class="line"><a id="l00723" name="l00723"></a><span class="lineno"> 723</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00724" name="l00724"></a><span class="lineno"> 724</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00725" name="l00725"></a><span class="lineno"> 725</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00726" name="l00726"></a><span class="lineno"> 726</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00727" name="l00727"></a><span class="lineno"> 727</span> </div>
|
|
<div class="line"><a id="l00729" name="l00729"></a><span class="lineno"><a class="line" href="group__ops.html#ga5528e80f5e8bad71e106a0cf9edd8920"> 729</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga5528e80f5e8bad71e106a0cf9edd8920">abs</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00730" name="l00730"></a><span class="lineno"> 730</span> </div>
|
|
<div class="line"><a id="l00732" name="l00732"></a><span class="lineno"><a class="line" href="group__ops.html#ga95d9a9425533b5ed1707eb00184dffc6"> 732</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga95d9a9425533b5ed1707eb00184dffc6">negative</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00733" name="l00733"></a><span class="lineno"><a class="line" href="group__ops.html#gade2eea48989f4caaf36e89f7bd2a8816"> 733</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gade2eea48989f4caaf36e89f7bd2a8816">operator-</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a);</div>
|
|
<div class="line"><a id="l00734" name="l00734"></a><span class="lineno"> 734</span> </div>
|
|
<div class="line"><a id="l00736" name="l00736"></a><span class="lineno"><a class="line" href="group__ops.html#ga20f1a1a8c0cd6206485f9363f3915faa"> 736</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga20f1a1a8c0cd6206485f9363f3915faa">sign</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00737" name="l00737"></a><span class="lineno"> 737</span> </div>
|
|
<div class="line"><a id="l00739" name="l00739"></a><span class="lineno"><a class="line" href="group__ops.html#gabca78d34ce93f0de2814e62225bb2a53"> 739</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gabca78d34ce93f0de2814e62225bb2a53">logical_not</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00740" name="l00740"></a><span class="lineno"> 740</span> </div>
|
|
<div class="line"><a id="l00742" name="l00742"></a><span class="lineno"><a class="line" href="group__ops.html#ga768977cda8d68cf23f464a6af9907876"> 742</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga768977cda8d68cf23f464a6af9907876">logical_and</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00743" name="l00743"></a><span class="lineno"><a class="line" href="group__ops.html#gaee1d774bb0843601d7a0a4257d616ae3"> 743</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaee1d774bb0843601d7a0a4257d616ae3">operator&&</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
|
<div class="line"><a id="l00744" name="l00744"></a><span class="lineno"> 744</span> </div>
|
|
<div class="line"><a id="l00746" name="l00746"></a><span class="lineno"><a class="line" href="group__ops.html#ga23768728e4dd070c917fbb0ed0d0c2ec"> 746</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga23768728e4dd070c917fbb0ed0d0c2ec">logical_or</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00747" name="l00747"></a><span class="lineno"><a class="line" href="group__ops.html#ga27af56a98270d4d76d139f0f9171b83a"> 747</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga27af56a98270d4d76d139f0f9171b83a">operator||</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
|
<div class="line"><a id="l00748" name="l00748"></a><span class="lineno"> 748</span> </div>
|
|
<div class="line"><a id="l00750" name="l00750"></a><span class="lineno"><a class="line" href="group__ops.html#ga4d29556bb93e2f66916116cf1f062b36"> 750</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga4d29556bb93e2f66916116cf1f062b36">reciprocal</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00751" name="l00751"></a><span class="lineno"> 751</span> </div>
|
|
<div class="line"><a id="l00753" name="l00753"></a><span class="lineno"><a class="line" href="group__ops.html#ga2d32d67cfd76785a72c43d89b94dc7d7"> 753</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga2d32d67cfd76785a72c43d89b94dc7d7">add</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00754" name="l00754"></a><span class="lineno"><a class="line" href="group__ops.html#ga26e5a043eaaaf066d1400adac9c11d0c"> 754</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga26e5a043eaaaf066d1400adac9c11d0c">operator+</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
|
<div class="line"><a id="l00755" name="l00755"></a><span class="lineno"> 755</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00756" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00756" name="l00756"></a><span class="lineno"><a class="line" href="group__ops.html#ga7d0ec8d01e7cefa6a6b25f11876761b5"> 756</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga26e5a043eaaaf066d1400adac9c11d0c">operator+</a>(T a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b) {</div>
|
|
<div class="line"><a id="l00757" name="l00757"></a><span class="lineno"> 757</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga2d32d67cfd76785a72c43d89b94dc7d7">add</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(a), b);</div>
|
|
<div class="line"><a id="l00758" name="l00758"></a><span class="lineno"> 758</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00759" name="l00759"></a><span class="lineno"> 759</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00760" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00760" name="l00760"></a><span class="lineno"><a class="line" href="group__ops.html#ga7cc080a4f9d4a667f2099aa0dbfefadd"> 760</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga26e5a043eaaaf066d1400adac9c11d0c">operator+</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, T b) {</div>
|
|
<div class="line"><a id="l00761" name="l00761"></a><span class="lineno"> 761</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga2d32d67cfd76785a72c43d89b94dc7d7">add</a>(a, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(b));</div>
|
|
<div class="line"><a id="l00762" name="l00762"></a><span class="lineno"> 762</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00763" name="l00763"></a><span class="lineno"> 763</span> </div>
|
|
<div class="line"><a id="l00765" name="l00765"></a><span class="lineno"><a class="line" href="group__ops.html#ga196c240d3d0fcbb4713802c485e15133"> 765</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga196c240d3d0fcbb4713802c485e15133">subtract</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00766" name="l00766"></a><span class="lineno"><a class="line" href="group__ops.html#ga0c7f3cb36d4ca516c7a33142f88b9181"> 766</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gade2eea48989f4caaf36e89f7bd2a8816">operator-</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
|
<div class="line"><a id="l00767" name="l00767"></a><span class="lineno"> 767</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00768" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00768" name="l00768"></a><span class="lineno"><a class="line" href="group__ops.html#gae68d3d0691ba951501218e98439f3465"> 768</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gade2eea48989f4caaf36e89f7bd2a8816">operator-</a>(T a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b) {</div>
|
|
<div class="line"><a id="l00769" name="l00769"></a><span class="lineno"> 769</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga196c240d3d0fcbb4713802c485e15133">subtract</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(a), b);</div>
|
|
<div class="line"><a id="l00770" name="l00770"></a><span class="lineno"> 770</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00771" name="l00771"></a><span class="lineno"> 771</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00772" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00772" name="l00772"></a><span class="lineno"><a class="line" href="group__ops.html#gaf5e5d882c51ad0a0ea315c274d5439b2"> 772</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gade2eea48989f4caaf36e89f7bd2a8816">operator-</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, T b) {</div>
|
|
<div class="line"><a id="l00773" name="l00773"></a><span class="lineno"> 773</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga196c240d3d0fcbb4713802c485e15133">subtract</a>(a, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(b));</div>
|
|
<div class="line"><a id="l00774" name="l00774"></a><span class="lineno"> 774</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00775" name="l00775"></a><span class="lineno"> 775</span> </div>
|
|
<div class="line"><a id="l00777" name="l00777"></a><span class="lineno"><a class="line" href="group__ops.html#gaf57392e641640b5d06e4c99518391c38"> 777</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaf57392e641640b5d06e4c99518391c38">multiply</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00778" name="l00778"></a><span class="lineno"><a class="line" href="group__ops.html#ga26c33f5cdb6fc10d272acd6e208034e0"> 778</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga26c33f5cdb6fc10d272acd6e208034e0">operator*</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
|
<div class="line"><a id="l00779" name="l00779"></a><span class="lineno"> 779</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00780" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00780" name="l00780"></a><span class="lineno"><a class="line" href="group__ops.html#gac22a67f7de797b1ae59029843cbdcab6"> 780</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga26c33f5cdb6fc10d272acd6e208034e0">operator*</a>(T a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b) {</div>
|
|
<div class="line"><a id="l00781" name="l00781"></a><span class="lineno"> 781</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gaf57392e641640b5d06e4c99518391c38">multiply</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(a), b);</div>
|
|
<div class="line"><a id="l00782" name="l00782"></a><span class="lineno"> 782</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00783" name="l00783"></a><span class="lineno"> 783</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00784" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00784" name="l00784"></a><span class="lineno"><a class="line" href="group__ops.html#ga6f2369ed5fae8ff9b1528670a004dde2"> 784</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga26c33f5cdb6fc10d272acd6e208034e0">operator*</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, T b) {</div>
|
|
<div class="line"><a id="l00785" name="l00785"></a><span class="lineno"> 785</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gaf57392e641640b5d06e4c99518391c38">multiply</a>(a, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(b));</div>
|
|
<div class="line"><a id="l00786" name="l00786"></a><span class="lineno"> 786</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00787" name="l00787"></a><span class="lineno"> 787</span> </div>
|
|
<div class="line"><a id="l00789" name="l00789"></a><span class="lineno"><a class="line" href="group__ops.html#ga77472dd06cfa7a30a42e4fd927bd859f"> 789</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga77472dd06cfa7a30a42e4fd927bd859f">divide</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00790" name="l00790"></a><span class="lineno"><a class="line" href="group__ops.html#gaeedf77f722b394429f1a7f6c367883bf"> 790</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaeedf77f722b394429f1a7f6c367883bf">operator/</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
|
<div class="line"><a id="l00791" name="l00791"></a><span class="lineno"><a class="line" href="group__ops.html#ga7366ec7f453be2a4dc449f0faa1bf554"> 791</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaeedf77f722b394429f1a7f6c367883bf">operator/</a>(<span class="keywordtype">double</span> a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
|
<div class="line"><a id="l00792" name="l00792"></a><span class="lineno"><a class="line" href="group__ops.html#gadfb324ae9b4feb2c7ea0ac6ade639f38"> 792</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaeedf77f722b394429f1a7f6c367883bf">operator/</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">double</span> b);</div>
|
|
<div class="line"><a id="l00793" name="l00793"></a><span class="lineno"> 793</span> </div>
|
|
<div class="line"><a id="l00795" name="l00795"></a><span class="lineno"> 795</span>std::vector<array></div>
|
|
<div class="line"><a id="l00796" name="l00796"></a><span class="lineno"><a class="line" href="group__ops.html#gaa30ebc0a8376dbc3f7e46a47052b5894"> 796</a></span><a class="code hl_function" href="group__ops.html#gaa30ebc0a8376dbc3f7e46a47052b5894">divmod</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00797" name="l00797"></a><span class="lineno"> 797</span> </div>
|
|
<div class="line"><a id="l00799" name="l00799"></a><span class="lineno"><a class="line" href="group__ops.html#ga05b4c6054d028107869511f927da01cd"> 799</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga05b4c6054d028107869511f927da01cd">floor_divide</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00800" name="l00800"></a><span class="lineno"> 800</span> </div>
|
|
<div class="line"><a id="l00802" name="l00802"></a><span class="lineno"><a class="line" href="group__ops.html#ga99f5c904f724156a814d7817188351d2"> 802</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga99f5c904f724156a814d7817188351d2">remainder</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00803" name="l00803"></a><span class="lineno"><a class="line" href="group__ops.html#gab3bfbf82b1e4de7b00bbcf1a2255fbde"> 803</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gab3bfbf82b1e4de7b00bbcf1a2255fbde">operator%</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
|
<div class="line"><a id="l00804" name="l00804"></a><span class="lineno"> 804</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00805" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00805" name="l00805"></a><span class="lineno"><a class="line" href="group__ops.html#ga50817666f0b82afcbf4a123486af9908"> 805</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gab3bfbf82b1e4de7b00bbcf1a2255fbde">operator%</a>(T a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b) {</div>
|
|
<div class="line"><a id="l00806" name="l00806"></a><span class="lineno"> 806</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga99f5c904f724156a814d7817188351d2">remainder</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(a), b);</div>
|
|
<div class="line"><a id="l00807" name="l00807"></a><span class="lineno"> 807</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00808" name="l00808"></a><span class="lineno"> 808</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00809" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00809" name="l00809"></a><span class="lineno"><a class="line" href="group__ops.html#ga46c01daa07433542a477d216e13a8480"> 809</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gab3bfbf82b1e4de7b00bbcf1a2255fbde">operator%</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, T b) {</div>
|
|
<div class="line"><a id="l00810" name="l00810"></a><span class="lineno"> 810</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga99f5c904f724156a814d7817188351d2">remainder</a>(a, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(b));</div>
|
|
<div class="line"><a id="l00811" name="l00811"></a><span class="lineno"> 811</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00812" name="l00812"></a><span class="lineno"> 812</span> </div>
|
|
<div class="line"><a id="l00814" name="l00814"></a><span class="lineno"><a class="line" href="group__ops.html#ga7ade2ea305e2e4219c3609443fb5db8d"> 814</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7ade2ea305e2e4219c3609443fb5db8d">maximum</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00815" name="l00815"></a><span class="lineno"> 815</span> </div>
|
|
<div class="line"><a id="l00817" name="l00817"></a><span class="lineno"><a class="line" href="group__ops.html#ga49ba00c090f81f331c91b0c97040bce0"> 817</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga49ba00c090f81f331c91b0c97040bce0">minimum</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00818" name="l00818"></a><span class="lineno"> 818</span> </div>
|
|
<div class="line"><a id="l00820" name="l00820"></a><span class="lineno"><a class="line" href="group__ops.html#ga8d656904aa2690b60955ae745aecfc30"> 820</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga8d656904aa2690b60955ae745aecfc30">floor</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00821" name="l00821"></a><span class="lineno"> 821</span> </div>
|
|
<div class="line"><a id="l00823" name="l00823"></a><span class="lineno"><a class="line" href="group__ops.html#ga1404ecceff83fd9b9139b7520f55e096"> 823</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga1404ecceff83fd9b9139b7520f55e096">ceil</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00824" name="l00824"></a><span class="lineno"> 824</span> </div>
|
|
<div class="line"><a id="l00826" name="l00826"></a><span class="lineno"><a class="line" href="group__ops.html#ga1234e4c39cfa79f19d4bdb5b8ea4d45e"> 826</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga1234e4c39cfa79f19d4bdb5b8ea4d45e">square</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00827" name="l00827"></a><span class="lineno"> 827</span> </div>
|
|
<div class="line"><a id="l00829" name="l00829"></a><span class="lineno"><a class="line" href="group__ops.html#ga8a3b04e23e347d99ecf411fd6f4e5125"> 829</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga8a3b04e23e347d99ecf411fd6f4e5125">exp</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00830" name="l00830"></a><span class="lineno"> 830</span> </div>
|
|
<div class="line"><a id="l00832" name="l00832"></a><span class="lineno"><a class="line" href="group__ops.html#gaebf0a73ad3732fba39df37826c235692"> 832</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaebf0a73ad3732fba39df37826c235692">sin</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00833" name="l00833"></a><span class="lineno"> 833</span> </div>
|
|
<div class="line"><a id="l00835" name="l00835"></a><span class="lineno"><a class="line" href="group__ops.html#ga39dfdf72b556012aa35ff27a94116e74"> 835</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga39dfdf72b556012aa35ff27a94116e74">cos</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00836" name="l00836"></a><span class="lineno"> 836</span> </div>
|
|
<div class="line"><a id="l00838" name="l00838"></a><span class="lineno"><a class="line" href="group__ops.html#ga3f10e89a4bcb1a8fa44fb33b8d1176a5"> 838</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga3f10e89a4bcb1a8fa44fb33b8d1176a5">tan</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00839" name="l00839"></a><span class="lineno"> 839</span> </div>
|
|
<div class="line"><a id="l00841" name="l00841"></a><span class="lineno"><a class="line" href="group__ops.html#ga8770e8c8f23f13343911f4c9d6e1c619"> 841</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga8770e8c8f23f13343911f4c9d6e1c619">arcsin</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00842" name="l00842"></a><span class="lineno"> 842</span> </div>
|
|
<div class="line"><a id="l00844" name="l00844"></a><span class="lineno"><a class="line" href="group__ops.html#ga08bec7cb10c84466487b507fc5bf9776"> 844</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga08bec7cb10c84466487b507fc5bf9776">arccos</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00845" name="l00845"></a><span class="lineno"> 845</span> </div>
|
|
<div class="line"><a id="l00847" name="l00847"></a><span class="lineno"><a class="line" href="group__ops.html#gaa041f3f070e68f4946db07516b7d092e"> 847</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaa041f3f070e68f4946db07516b7d092e">arctan</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00848" name="l00848"></a><span class="lineno"> 848</span> </div>
|
|
<div class="line"><a id="l00850" name="l00850"></a><span class="lineno"><a class="line" href="group__ops.html#ga6caba9c92b5989123501f909cc7da354"> 850</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga6caba9c92b5989123501f909cc7da354">arctan2</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00851" name="l00851"></a><span class="lineno"> 851</span> </div>
|
|
<div class="line"><a id="l00853" name="l00853"></a><span class="lineno"><a class="line" href="group__ops.html#gaf532375c6563dbd6e329bdedf0224dd7"> 853</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaf532375c6563dbd6e329bdedf0224dd7">sinh</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00854" name="l00854"></a><span class="lineno"> 854</span> </div>
|
|
<div class="line"><a id="l00856" name="l00856"></a><span class="lineno"><a class="line" href="group__ops.html#ga2181b71cda88007a3092be4795ff0715"> 856</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga2181b71cda88007a3092be4795ff0715">cosh</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00857" name="l00857"></a><span class="lineno"> 857</span> </div>
|
|
<div class="line"><a id="l00859" name="l00859"></a><span class="lineno"><a class="line" href="group__ops.html#ga5efb19aa0dfa42d8a3d5e1dfd569cd6d"> 859</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga5efb19aa0dfa42d8a3d5e1dfd569cd6d">tanh</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00860" name="l00860"></a><span class="lineno"> 860</span> </div>
|
|
<div class="line"><a id="l00862" name="l00862"></a><span class="lineno"><a class="line" href="group__ops.html#gac62e2cedc49ef2c90dd8584000317450"> 862</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac62e2cedc49ef2c90dd8584000317450">arcsinh</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00863" name="l00863"></a><span class="lineno"> 863</span> </div>
|
|
<div class="line"><a id="l00865" name="l00865"></a><span class="lineno"><a class="line" href="group__ops.html#gaafafcfcebdf7248679c8543d0c0497e5"> 865</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaafafcfcebdf7248679c8543d0c0497e5">arccosh</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00866" name="l00866"></a><span class="lineno"> 866</span> </div>
|
|
<div class="line"><a id="l00868" name="l00868"></a><span class="lineno"><a class="line" href="group__ops.html#gab46a35925a04c5a9d2ec7898ee55358e"> 868</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gab46a35925a04c5a9d2ec7898ee55358e">arctanh</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00869" name="l00869"></a><span class="lineno"> 869</span> </div>
|
|
<div class="line"><a id="l00871" name="l00871"></a><span class="lineno"><a class="line" href="group__ops.html#ga3a70569b50e1083c5ded199d73fb960c"> 871</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga3a70569b50e1083c5ded199d73fb960c">degrees</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00872" name="l00872"></a><span class="lineno"> 872</span> </div>
|
|
<div class="line"><a id="l00874" name="l00874"></a><span class="lineno"><a class="line" href="group__ops.html#ga07aa8059adba5b9a8818027b8aafd31e"> 874</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga07aa8059adba5b9a8818027b8aafd31e">radians</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00875" name="l00875"></a><span class="lineno"> 875</span> </div>
|
|
<div class="line"><a id="l00877" name="l00877"></a><span class="lineno"><a class="line" href="group__ops.html#ga6fb22d4926133573e430fcc92f4eef31"> 877</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga6fb22d4926133573e430fcc92f4eef31">log</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00878" name="l00878"></a><span class="lineno"> 878</span> </div>
|
|
<div class="line"><a id="l00880" name="l00880"></a><span class="lineno"><a class="line" href="group__ops.html#ga144228d7222d15af3a135b8b0f3fa21b"> 880</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga144228d7222d15af3a135b8b0f3fa21b">log2</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00881" name="l00881"></a><span class="lineno"> 881</span> </div>
|
|
<div class="line"><a id="l00883" name="l00883"></a><span class="lineno"><a class="line" href="group__ops.html#ga1fdcc7fc8819caf2e6f1c327ed4e9b9e"> 883</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga1fdcc7fc8819caf2e6f1c327ed4e9b9e">log10</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00884" name="l00884"></a><span class="lineno"> 884</span> </div>
|
|
<div class="line"><a id="l00886" name="l00886"></a><span class="lineno"><a class="line" href="group__ops.html#ga20a1f4270c35b0fa544f5105a87a1604"> 886</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga20a1f4270c35b0fa544f5105a87a1604">log1p</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00887" name="l00887"></a><span class="lineno"> 887</span> </div>
|
|
<div class="line"><a id="l00889" name="l00889"></a><span class="lineno"><a class="line" href="group__ops.html#gaf985df6609c6bd75a14a844655d89eaa"> 889</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaf985df6609c6bd75a14a844655d89eaa">logaddexp</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00890" name="l00890"></a><span class="lineno"> 890</span> </div>
|
|
<div class="line"><a id="l00892" name="l00892"></a><span class="lineno"><a class="line" href="group__ops.html#ga708abf8f79609cd6831db7c38cafac0e"> 892</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga708abf8f79609cd6831db7c38cafac0e">sigmoid</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00893" name="l00893"></a><span class="lineno"> 893</span> </div>
|
|
<div class="line"><a id="l00895" name="l00895"></a><span class="lineno"><a class="line" href="group__ops.html#ga292a335240fd5d6d625fb7a340ff5eb0"> 895</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga292a335240fd5d6d625fb7a340ff5eb0">erf</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00896" name="l00896"></a><span class="lineno"> 896</span> </div>
|
|
<div class="line"><a id="l00898" name="l00898"></a><span class="lineno"><a class="line" href="group__ops.html#ga76fb9062c64264e34d2e07013390557c"> 898</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga76fb9062c64264e34d2e07013390557c">erfinv</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00899" name="l00899"></a><span class="lineno"> 899</span> </div>
|
|
<div class="line"><a id="l00901" name="l00901"></a><span class="lineno"><a class="line" href="group__ops.html#ga54ca54f06bfb2be15b163a5209e2a0f0"> 901</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga54ca54f06bfb2be15b163a5209e2a0f0">expm1</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00902" name="l00902"></a><span class="lineno"> 902</span> </div>
|
|
<div class="line"><a id="l00904" name="l00904"></a><span class="lineno"><a class="line" href="group__ops.html#ga36bc28f1deb2fe668ca9ae1e447b6b1f"> 904</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga36bc28f1deb2fe668ca9ae1e447b6b1f">stop_gradient</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00905" name="l00905"></a><span class="lineno"> 905</span> </div>
|
|
<div class="line"><a id="l00907" name="l00907"></a><span class="lineno"><a class="line" href="group__ops.html#ga2d74d43f007a069384e89d8416525331"> 907</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga2d74d43f007a069384e89d8416525331">round</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> decimals, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00908" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00908" name="l00908"></a><span class="lineno"><a class="line" href="group__ops.html#gaf18fb7e98bf8cf3b7fbc5e64c988a95b"> 908</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga2d74d43f007a069384e89d8416525331">round</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00909" name="l00909"></a><span class="lineno"> 909</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga2d74d43f007a069384e89d8416525331">round</a>(a, 0, s);</div>
|
|
<div class="line"><a id="l00910" name="l00910"></a><span class="lineno"> 910</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00911" name="l00911"></a><span class="lineno"> 911</span> </div>
|
|
<div class="line"><a id="l00913" name="l00913"></a><span class="lineno"><a class="line" href="group__ops.html#ga753d59f5a9f5f2362865ee83b4dced2a"> 913</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga753d59f5a9f5f2362865ee83b4dced2a">matmul</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00914" name="l00914"></a><span class="lineno"> 914</span> </div>
|
|
<div class="line"><a id="l00916" name="l00916"></a><span class="lineno"><a class="line" href="group__ops.html#gab6e7f655a9ff15350ca5379692f9d444"> 916</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gab6e7f655a9ff15350ca5379692f9d444">gather</a>(</div>
|
|
<div class="line"><a id="l00917" name="l00917"></a><span class="lineno"> 917</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00918" name="l00918"></a><span class="lineno"> 918</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
|
<div class="line"><a id="l00919" name="l00919"></a><span class="lineno"> 919</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00920" name="l00920"></a><span class="lineno"> 920</span> <span class="keyword">const</span> std::vector<int>& slice_sizes,</div>
|
|
<div class="line"><a id="l00921" name="l00921"></a><span class="lineno"> 921</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00922" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00922" name="l00922"></a><span class="lineno"><a class="line" href="group__ops.html#gadb4337ca5d4f88fe9e7c083bc478158b"> 922</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gab6e7f655a9ff15350ca5379692f9d444">gather</a>(</div>
|
|
<div class="line"><a id="l00923" name="l00923"></a><span class="lineno"> 923</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00924" name="l00924"></a><span class="lineno"> 924</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& indices,</div>
|
|
<div class="line"><a id="l00925" name="l00925"></a><span class="lineno"> 925</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00926" name="l00926"></a><span class="lineno"> 926</span> <span class="keyword">const</span> std::vector<int>& slice_sizes,</div>
|
|
<div class="line"><a id="l00927" name="l00927"></a><span class="lineno"> 927</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00928" name="l00928"></a><span class="lineno"> 928</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gab6e7f655a9ff15350ca5379692f9d444">gather</a>(a, {indices}, std::vector<int>{axis}, slice_sizes, s);</div>
|
|
<div class="line"><a id="l00929" name="l00929"></a><span class="lineno"> 929</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00930" name="l00930"></a><span class="lineno"> 930</span> </div>
|
|
<div class="line"><a id="l00932" name="l00932"></a><span class="lineno"><a class="line" href="group__ops.html#gac2fc270882fcfa81eb8bd068cc0d86d7"> 932</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac2fc270882fcfa81eb8bd068cc0d86d7">take</a>(</div>
|
|
<div class="line"><a id="l00933" name="l00933"></a><span class="lineno"> 933</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00934" name="l00934"></a><span class="lineno"> 934</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& indices,</div>
|
|
<div class="line"><a id="l00935" name="l00935"></a><span class="lineno"> 935</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00936" name="l00936"></a><span class="lineno"> 936</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00937" name="l00937"></a><span class="lineno"><a class="line" href="group__ops.html#ga731af77b7de547a73336f91c829c40b4"> 937</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac2fc270882fcfa81eb8bd068cc0d86d7">take</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> index, <span class="keywordtype">int</span> axis, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00938" name="l00938"></a><span class="lineno"> 938</span> </div>
|
|
<div class="line"><a id="l00940" name="l00940"></a><span class="lineno"><a class="line" href="group__ops.html#ga45d0f423a5e030440ef753f36c5aabf1"> 940</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac2fc270882fcfa81eb8bd068cc0d86d7">take</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& indices, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00941" name="l00941"></a><span class="lineno"><a class="line" href="group__ops.html#ga7b3c9b4c8ee02dc23cfd7dcd855dbf20"> 941</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac2fc270882fcfa81eb8bd068cc0d86d7">take</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> index, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00942" name="l00942"></a><span class="lineno"> 942</span> </div>
|
|
<div class="line"><a id="l00944" name="l00944"></a><span class="lineno"><a class="line" href="group__ops.html#gae0a81d4f983e296a87302e36d65bfc76"> 944</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gae0a81d4f983e296a87302e36d65bfc76">take_along_axis</a>(</div>
|
|
<div class="line"><a id="l00945" name="l00945"></a><span class="lineno"> 945</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00946" name="l00946"></a><span class="lineno"> 946</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& indices,</div>
|
|
<div class="line"><a id="l00947" name="l00947"></a><span class="lineno"> 947</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00948" name="l00948"></a><span class="lineno"> 948</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00949" name="l00949"></a><span class="lineno"> 949</span> </div>
|
|
<div class="line"><a id="l00951" name="l00951"></a><span class="lineno"><a class="line" href="group__ops.html#ga8e0caebf43cd65bd40e4ce97922cd06b"> 951</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga8e0caebf43cd65bd40e4ce97922cd06b">put_along_axis</a>(</div>
|
|
<div class="line"><a id="l00952" name="l00952"></a><span class="lineno"> 952</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l00953" name="l00953"></a><span class="lineno"> 953</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& indices,</div>
|
|
<div class="line"><a id="l00954" name="l00954"></a><span class="lineno"> 954</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& values,</div>
|
|
<div class="line"><a id="l00955" name="l00955"></a><span class="lineno"> 955</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00956" name="l00956"></a><span class="lineno"> 956</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00957" name="l00957"></a><span class="lineno"> 957</span> </div>
|
|
<div class="line"><a id="l01056" name="l01056"></a><span class="lineno"><a class="line" href="group__ops.html#gad438be8f90bae9d37c6853b8f4225d61"> 1056</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gad438be8f90bae9d37c6853b8f4225d61">scatter</a>(</div>
|
|
<div class="line"><a id="l01057" name="l01057"></a><span class="lineno"> 1057</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01058" name="l01058"></a><span class="lineno"> 1058</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
|
<div class="line"><a id="l01059" name="l01059"></a><span class="lineno"> 1059</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
|
<div class="line"><a id="l01060" name="l01060"></a><span class="lineno"> 1060</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l01061" name="l01061"></a><span class="lineno"> 1061</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen01062" data-start="{" data-end="}">
|
|
<div class="line"><a id="l01062" name="l01062"></a><span class="lineno"><a class="line" href="group__ops.html#gac2c2b379a3ce959dbe1c4a68f112edfe"> 1062</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gad438be8f90bae9d37c6853b8f4225d61">scatter</a>(</div>
|
|
<div class="line"><a id="l01063" name="l01063"></a><span class="lineno"> 1063</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01064" name="l01064"></a><span class="lineno"> 1064</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& indices,</div>
|
|
<div class="line"><a id="l01065" name="l01065"></a><span class="lineno"> 1065</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
|
<div class="line"><a id="l01066" name="l01066"></a><span class="lineno"> 1066</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l01067" name="l01067"></a><span class="lineno"> 1067</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l01068" name="l01068"></a><span class="lineno"> 1068</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gad438be8f90bae9d37c6853b8f4225d61">scatter</a>(a, {indices}, updates, std::vector<int>{axis}, s);</div>
|
|
<div class="line"><a id="l01069" name="l01069"></a><span class="lineno"> 1069</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l01070" name="l01070"></a><span class="lineno"> 1070</span> </div>
|
|
<div class="line"><a id="l01072" name="l01072"></a><span class="lineno"><a class="line" href="group__ops.html#gacd14c2b5cfebf343fc2d672722f8d174"> 1072</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gacd14c2b5cfebf343fc2d672722f8d174">scatter_add</a>(</div>
|
|
<div class="line"><a id="l01073" name="l01073"></a><span class="lineno"> 1073</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01074" name="l01074"></a><span class="lineno"> 1074</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
|
<div class="line"><a id="l01075" name="l01075"></a><span class="lineno"> 1075</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
|
<div class="line"><a id="l01076" name="l01076"></a><span class="lineno"> 1076</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l01077" name="l01077"></a><span class="lineno"> 1077</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen01078" data-start="{" data-end="}">
|
|
<div class="line"><a id="l01078" name="l01078"></a><span class="lineno"><a class="line" href="group__ops.html#gac13318518e5703f1273c5366eb523a5a"> 1078</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gacd14c2b5cfebf343fc2d672722f8d174">scatter_add</a>(</div>
|
|
<div class="line"><a id="l01079" name="l01079"></a><span class="lineno"> 1079</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01080" name="l01080"></a><span class="lineno"> 1080</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& indices,</div>
|
|
<div class="line"><a id="l01081" name="l01081"></a><span class="lineno"> 1081</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
|
<div class="line"><a id="l01082" name="l01082"></a><span class="lineno"> 1082</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l01083" name="l01083"></a><span class="lineno"> 1083</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l01084" name="l01084"></a><span class="lineno"> 1084</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gacd14c2b5cfebf343fc2d672722f8d174">scatter_add</a>(a, {indices}, updates, std::vector<int>{axis}, s);</div>
|
|
<div class="line"><a id="l01085" name="l01085"></a><span class="lineno"> 1085</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l01086" name="l01086"></a><span class="lineno"> 1086</span> </div>
|
|
<div class="line"><a id="l01088" name="l01088"></a><span class="lineno"><a class="line" href="group__ops.html#ga3708b5bcb61e2c63d213c4ce6ad0ffc0"> 1088</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga3708b5bcb61e2c63d213c4ce6ad0ffc0">scatter_prod</a>(</div>
|
|
<div class="line"><a id="l01089" name="l01089"></a><span class="lineno"> 1089</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01090" name="l01090"></a><span class="lineno"> 1090</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
|
<div class="line"><a id="l01091" name="l01091"></a><span class="lineno"> 1091</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
|
<div class="line"><a id="l01092" name="l01092"></a><span class="lineno"> 1092</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l01093" name="l01093"></a><span class="lineno"> 1093</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen01094" data-start="{" data-end="}">
|
|
<div class="line"><a id="l01094" name="l01094"></a><span class="lineno"><a class="line" href="group__ops.html#gaf83c53c453faa9083ba27e4b97539339"> 1094</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga3708b5bcb61e2c63d213c4ce6ad0ffc0">scatter_prod</a>(</div>
|
|
<div class="line"><a id="l01095" name="l01095"></a><span class="lineno"> 1095</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01096" name="l01096"></a><span class="lineno"> 1096</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& indices,</div>
|
|
<div class="line"><a id="l01097" name="l01097"></a><span class="lineno"> 1097</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
|
<div class="line"><a id="l01098" name="l01098"></a><span class="lineno"> 1098</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l01099" name="l01099"></a><span class="lineno"> 1099</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l01100" name="l01100"></a><span class="lineno"> 1100</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga3708b5bcb61e2c63d213c4ce6ad0ffc0">scatter_prod</a>(a, {indices}, updates, std::vector<int>{axis}, s);</div>
|
|
<div class="line"><a id="l01101" name="l01101"></a><span class="lineno"> 1101</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l01102" name="l01102"></a><span class="lineno"> 1102</span> </div>
|
|
<div class="line"><a id="l01104" name="l01104"></a><span class="lineno"><a class="line" href="group__ops.html#ga05881a4157cd113c9392d168a79e6673"> 1104</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga05881a4157cd113c9392d168a79e6673">scatter_max</a>(</div>
|
|
<div class="line"><a id="l01105" name="l01105"></a><span class="lineno"> 1105</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01106" name="l01106"></a><span class="lineno"> 1106</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
|
<div class="line"><a id="l01107" name="l01107"></a><span class="lineno"> 1107</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
|
<div class="line"><a id="l01108" name="l01108"></a><span class="lineno"> 1108</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l01109" name="l01109"></a><span class="lineno"> 1109</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen01110" data-start="{" data-end="}">
|
|
<div class="line"><a id="l01110" name="l01110"></a><span class="lineno"><a class="line" href="group__ops.html#ga9adda5f9202bb3486e4d9e1114e3a56f"> 1110</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga05881a4157cd113c9392d168a79e6673">scatter_max</a>(</div>
|
|
<div class="line"><a id="l01111" name="l01111"></a><span class="lineno"> 1111</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01112" name="l01112"></a><span class="lineno"> 1112</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& indices,</div>
|
|
<div class="line"><a id="l01113" name="l01113"></a><span class="lineno"> 1113</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
|
<div class="line"><a id="l01114" name="l01114"></a><span class="lineno"> 1114</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l01115" name="l01115"></a><span class="lineno"> 1115</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l01116" name="l01116"></a><span class="lineno"> 1116</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga05881a4157cd113c9392d168a79e6673">scatter_max</a>(a, {indices}, updates, std::vector<int>{axis}, s);</div>
|
|
<div class="line"><a id="l01117" name="l01117"></a><span class="lineno"> 1117</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l01119" name="l01119"></a><span class="lineno"><a class="line" href="group__ops.html#ga0ca16b7579dfc899f3f7fd40245ba7c5"> 1119</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga0ca16b7579dfc899f3f7fd40245ba7c5">scatter_min</a>(</div>
|
|
<div class="line"><a id="l01120" name="l01120"></a><span class="lineno"> 1120</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01121" name="l01121"></a><span class="lineno"> 1121</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
|
<div class="line"><a id="l01122" name="l01122"></a><span class="lineno"> 1122</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
|
<div class="line"><a id="l01123" name="l01123"></a><span class="lineno"> 1123</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l01124" name="l01124"></a><span class="lineno"> 1124</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen01125" data-start="{" data-end="}">
|
|
<div class="line"><a id="l01125" name="l01125"></a><span class="lineno"><a class="line" href="group__ops.html#ga51fa762a997c243ca7a19e1ed3e83199"> 1125</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga0ca16b7579dfc899f3f7fd40245ba7c5">scatter_min</a>(</div>
|
|
<div class="line"><a id="l01126" name="l01126"></a><span class="lineno"> 1126</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01127" name="l01127"></a><span class="lineno"> 1127</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& indices,</div>
|
|
<div class="line"><a id="l01128" name="l01128"></a><span class="lineno"> 1128</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
|
<div class="line"><a id="l01129" name="l01129"></a><span class="lineno"> 1129</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l01130" name="l01130"></a><span class="lineno"> 1130</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l01131" name="l01131"></a><span class="lineno"> 1131</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga0ca16b7579dfc899f3f7fd40245ba7c5">scatter_min</a>(a, {indices}, updates, std::vector<int>{axis}, s);</div>
|
|
<div class="line"><a id="l01132" name="l01132"></a><span class="lineno"> 1132</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l01133" name="l01133"></a><span class="lineno"> 1133</span> </div>
|
|
<div class="line"><a id="l01135" name="l01135"></a><span class="lineno"><a class="line" href="group__ops.html#ga297f853b3d90ec8ae81263977ba2ddb1"> 1135</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga297f853b3d90ec8ae81263977ba2ddb1">sqrt</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01136" name="l01136"></a><span class="lineno"> 1136</span> </div>
|
|
<div class="line"><a id="l01138" name="l01138"></a><span class="lineno"><a class="line" href="group__ops.html#ga102f23aa0b0c3d3296a321c694617aa1"> 1138</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga102f23aa0b0c3d3296a321c694617aa1">rsqrt</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01139" name="l01139"></a><span class="lineno"> 1139</span> </div>
|
|
<div class="line"><a id="l01141" name="l01141"></a><span class="lineno"><a class="line" href="group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6"> 1141</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6">softmax</a>(</div>
|
|
<div class="line"><a id="l01142" name="l01142"></a><span class="lineno"> 1142</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01143" name="l01143"></a><span class="lineno"> 1143</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l01144" name="l01144"></a><span class="lineno"> 1144</span> <span class="keywordtype">bool</span> precise = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l01145" name="l01145"></a><span class="lineno"> 1145</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01146" name="l01146"></a><span class="lineno"> 1146</span> </div>
|
|
<div class="line"><a id="l01148" name="l01148"></a><span class="lineno"><a class="line" href="group__ops.html#ga1ae3614d07d873892a530d14c3857d0b"> 1148</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6">softmax</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">bool</span> precise = <span class="keyword">false</span>, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01149" name="l01149"></a><span class="lineno"> 1149</span> </div>
|
|
<div class="line"><a id="l01151" name="l01151"></a><span class="lineno"> 1151</span><span class="keyword">inline</span> array</div>
|
|
<div class="foldopen" id="foldopen01152" data-start="{" data-end="}">
|
|
<div class="line"><a id="l01152" name="l01152"></a><span class="lineno"><a class="line" href="group__ops.html#ga06f570d73716a24303e6de3aaba4457b"> 1152</a></span><a class="code hl_function" href="group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6">softmax</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> axis, <span class="keywordtype">bool</span> precise = <span class="keyword">false</span>, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l01153" name="l01153"></a><span class="lineno"> 1153</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6">softmax</a>(a, std::vector<int>{axis}, precise, s);</div>
|
|
<div class="line"><a id="l01154" name="l01154"></a><span class="lineno"> 1154</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l01155" name="l01155"></a><span class="lineno"> 1155</span> </div>
|
|
<div class="line"><a id="l01157" name="l01157"></a><span class="lineno"><a class="line" href="group__ops.html#ga7972058715c26559dff9c9ae2a3ef76d"> 1157</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7972058715c26559dff9c9ae2a3ef76d">power</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01158" name="l01158"></a><span class="lineno"> 1158</span> </div>
|
|
<div class="line"><a id="l01160" name="l01160"></a><span class="lineno"><a class="line" href="group__ops.html#gaddc825a5c173e195ab0fda83ad630420"> 1160</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaddc825a5c173e195ab0fda83ad630420">cumsum</a>(</div>
|
|
<div class="line"><a id="l01161" name="l01161"></a><span class="lineno"> 1161</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01162" name="l01162"></a><span class="lineno"> 1162</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l01163" name="l01163"></a><span class="lineno"> 1163</span> <span class="keywordtype">bool</span> reverse = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l01164" name="l01164"></a><span class="lineno"> 1164</span> <span class="keywordtype">bool</span> inclusive = <span class="keyword">true</span>,</div>
|
|
<div class="line"><a id="l01165" name="l01165"></a><span class="lineno"> 1165</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01166" name="l01166"></a><span class="lineno"> 1166</span> </div>
|
|
<div class="line"><a id="l01168" name="l01168"></a><span class="lineno"><a class="line" href="group__ops.html#ga0d71dfbc14ef3ed564b0c5ee26af680f"> 1168</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga0d71dfbc14ef3ed564b0c5ee26af680f">cumprod</a>(</div>
|
|
<div class="line"><a id="l01169" name="l01169"></a><span class="lineno"> 1169</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01170" name="l01170"></a><span class="lineno"> 1170</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l01171" name="l01171"></a><span class="lineno"> 1171</span> <span class="keywordtype">bool</span> reverse = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l01172" name="l01172"></a><span class="lineno"> 1172</span> <span class="keywordtype">bool</span> inclusive = <span class="keyword">true</span>,</div>
|
|
<div class="line"><a id="l01173" name="l01173"></a><span class="lineno"> 1173</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01174" name="l01174"></a><span class="lineno"> 1174</span> </div>
|
|
<div class="line"><a id="l01176" name="l01176"></a><span class="lineno"><a class="line" href="group__ops.html#gaee37cac8476e8f8d666bcded5bc59143"> 1176</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaee37cac8476e8f8d666bcded5bc59143">cummax</a>(</div>
|
|
<div class="line"><a id="l01177" name="l01177"></a><span class="lineno"> 1177</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01178" name="l01178"></a><span class="lineno"> 1178</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l01179" name="l01179"></a><span class="lineno"> 1179</span> <span class="keywordtype">bool</span> reverse = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l01180" name="l01180"></a><span class="lineno"> 1180</span> <span class="keywordtype">bool</span> inclusive = <span class="keyword">true</span>,</div>
|
|
<div class="line"><a id="l01181" name="l01181"></a><span class="lineno"> 1181</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01182" name="l01182"></a><span class="lineno"> 1182</span> </div>
|
|
<div class="line"><a id="l01184" name="l01184"></a><span class="lineno"><a class="line" href="group__ops.html#ga19c1bf6929fe8d66b9cd408946aea6a8"> 1184</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga19c1bf6929fe8d66b9cd408946aea6a8">cummin</a>(</div>
|
|
<div class="line"><a id="l01185" name="l01185"></a><span class="lineno"> 1185</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01186" name="l01186"></a><span class="lineno"> 1186</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l01187" name="l01187"></a><span class="lineno"> 1187</span> <span class="keywordtype">bool</span> reverse = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l01188" name="l01188"></a><span class="lineno"> 1188</span> <span class="keywordtype">bool</span> inclusive = <span class="keyword">true</span>,</div>
|
|
<div class="line"><a id="l01189" name="l01189"></a><span class="lineno"> 1189</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01190" name="l01190"></a><span class="lineno"> 1190</span> </div>
|
|
<div class="line"><a id="l01192" name="l01192"></a><span class="lineno"><a class="line" href="group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e"> 1192</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e">conv_general</a>(</div>
|
|
<div class="line"><a id="l01193" name="l01193"></a><span class="lineno"> 1193</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> input,</div>
|
|
<div class="line"><a id="l01194" name="l01194"></a><span class="lineno"> 1194</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> weight,</div>
|
|
<div class="line"><a id="l01195" name="l01195"></a><span class="lineno"> 1195</span> std::vector<int> stride = {},</div>
|
|
<div class="line"><a id="l01196" name="l01196"></a><span class="lineno"> 1196</span> std::vector<int> padding_lo = {},</div>
|
|
<div class="line"><a id="l01197" name="l01197"></a><span class="lineno"> 1197</span> std::vector<int> padding_hi = {},</div>
|
|
<div class="line"><a id="l01198" name="l01198"></a><span class="lineno"> 1198</span> std::vector<int> kernel_dilation = {},</div>
|
|
<div class="line"><a id="l01199" name="l01199"></a><span class="lineno"> 1199</span> std::vector<int> input_dilation = {},</div>
|
|
<div class="line"><a id="l01200" name="l01200"></a><span class="lineno"> 1200</span> <span class="keywordtype">int</span> groups = 1,</div>
|
|
<div class="line"><a id="l01201" name="l01201"></a><span class="lineno"> 1201</span> <span class="keywordtype">bool</span> flip = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l01202" name="l01202"></a><span class="lineno"> 1202</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01203" name="l01203"></a><span class="lineno"> 1203</span> </div>
|
|
<div class="foldopen" id="foldopen01205" data-start="{" data-end="}">
|
|
<div class="line"><a id="l01205" name="l01205"></a><span class="lineno"><a class="line" href="group__ops.html#gab59f89942cd1efaadffe9e8762e3c99d"> 1205</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e">conv_general</a>(</div>
|
|
<div class="line"><a id="l01206" name="l01206"></a><span class="lineno"> 1206</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& input,</div>
|
|
<div class="line"><a id="l01207" name="l01207"></a><span class="lineno"> 1207</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& weight,</div>
|
|
<div class="line"><a id="l01208" name="l01208"></a><span class="lineno"> 1208</span> std::vector<int> stride = {},</div>
|
|
<div class="line"><a id="l01209" name="l01209"></a><span class="lineno"> 1209</span> std::vector<int> padding = {},</div>
|
|
<div class="line"><a id="l01210" name="l01210"></a><span class="lineno"> 1210</span> std::vector<int> kernel_dilation = {},</div>
|
|
<div class="line"><a id="l01211" name="l01211"></a><span class="lineno"> 1211</span> std::vector<int> input_dilation = {},</div>
|
|
<div class="line"><a id="l01212" name="l01212"></a><span class="lineno"> 1212</span> <span class="keywordtype">int</span> groups = 1,</div>
|
|
<div class="line"><a id="l01213" name="l01213"></a><span class="lineno"> 1213</span> <span class="keywordtype">bool</span> flip = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l01214" name="l01214"></a><span class="lineno"> 1214</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l01215" name="l01215"></a><span class="lineno"> 1215</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e">conv_general</a>(</div>
|
|
<div class="line"><a id="l01216" name="l01216"></a><span class="lineno"> 1216</span> <span class="comment">/* const array& input = */</span> input,</div>
|
|
<div class="line"><a id="l01217" name="l01217"></a><span class="lineno"> 1217</span> <span class="comment">/* const array& weight = */</span> weight,</div>
|
|
<div class="line"><a id="l01218" name="l01218"></a><span class="lineno"> 1218</span> <span class="comment">/* std::vector<int> stride = */</span> stride,</div>
|
|
<div class="line"><a id="l01219" name="l01219"></a><span class="lineno"> 1219</span> <span class="comment">/* std::vector<int> padding_lo = */</span> padding,</div>
|
|
<div class="line"><a id="l01220" name="l01220"></a><span class="lineno"> 1220</span> <span class="comment">/* std::vector<int> padding_hi = */</span> padding,</div>
|
|
<div class="line"><a id="l01221" name="l01221"></a><span class="lineno"> 1221</span> <span class="comment">/* std::vector<int> kernel_dilation = */</span> kernel_dilation,</div>
|
|
<div class="line"><a id="l01222" name="l01222"></a><span class="lineno"> 1222</span> <span class="comment">/* std::vector<int> input_dilation = */</span> input_dilation,</div>
|
|
<div class="line"><a id="l01223" name="l01223"></a><span class="lineno"> 1223</span> <span class="comment">/* int groups = */</span> groups,</div>
|
|
<div class="line"><a id="l01224" name="l01224"></a><span class="lineno"> 1224</span> <span class="comment">/* bool flip = */</span> flip,</div>
|
|
<div class="line"><a id="l01225" name="l01225"></a><span class="lineno"> 1225</span> <span class="comment">/* StreamOrDevice s = */</span> s);</div>
|
|
<div class="line"><a id="l01226" name="l01226"></a><span class="lineno"> 1226</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l01227" name="l01227"></a><span class="lineno"> 1227</span> </div>
|
|
<div class="line"><a id="l01229" name="l01229"></a><span class="lineno"><a class="line" href="group__ops.html#ga30d47e08093c03a3676f235f9f559411"> 1229</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga30d47e08093c03a3676f235f9f559411">conv1d</a>(</div>
|
|
<div class="line"><a id="l01230" name="l01230"></a><span class="lineno"> 1230</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& input,</div>
|
|
<div class="line"><a id="l01231" name="l01231"></a><span class="lineno"> 1231</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& weight,</div>
|
|
<div class="line"><a id="l01232" name="l01232"></a><span class="lineno"> 1232</span> <span class="keywordtype">int</span> stride = 1,</div>
|
|
<div class="line"><a id="l01233" name="l01233"></a><span class="lineno"> 1233</span> <span class="keywordtype">int</span> padding = 0,</div>
|
|
<div class="line"><a id="l01234" name="l01234"></a><span class="lineno"> 1234</span> <span class="keywordtype">int</span> dilation = 1,</div>
|
|
<div class="line"><a id="l01235" name="l01235"></a><span class="lineno"> 1235</span> <span class="keywordtype">int</span> groups = 1,</div>
|
|
<div class="line"><a id="l01236" name="l01236"></a><span class="lineno"> 1236</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01237" name="l01237"></a><span class="lineno"> 1237</span> </div>
|
|
<div class="line"><a id="l01239" name="l01239"></a><span class="lineno"><a class="line" href="group__ops.html#ga73b02833229678786e7f302d458d5a83"> 1239</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga73b02833229678786e7f302d458d5a83">conv2d</a>(</div>
|
|
<div class="line"><a id="l01240" name="l01240"></a><span class="lineno"> 1240</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& input,</div>
|
|
<div class="line"><a id="l01241" name="l01241"></a><span class="lineno"> 1241</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& weight,</div>
|
|
<div class="line"><a id="l01242" name="l01242"></a><span class="lineno"> 1242</span> <span class="keyword">const</span> std::pair<int, int>& stride = {1, 1},</div>
|
|
<div class="line"><a id="l01243" name="l01243"></a><span class="lineno"> 1243</span> <span class="keyword">const</span> std::pair<int, int>& padding = {0, 0},</div>
|
|
<div class="line"><a id="l01244" name="l01244"></a><span class="lineno"> 1244</span> <span class="keyword">const</span> std::pair<int, int>& dilation = {1, 1},</div>
|
|
<div class="line"><a id="l01245" name="l01245"></a><span class="lineno"> 1245</span> <span class="keywordtype">int</span> groups = 1,</div>
|
|
<div class="line"><a id="l01246" name="l01246"></a><span class="lineno"> 1246</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01247" name="l01247"></a><span class="lineno"> 1247</span> </div>
|
|
<div class="line"><a id="l01249" name="l01249"></a><span class="lineno"><a class="line" href="group__ops.html#ga6e9907d2f14dc4803e4306b3dbc4b3ca"> 1249</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga6e9907d2f14dc4803e4306b3dbc4b3ca">conv3d</a>(</div>
|
|
<div class="line"><a id="l01250" name="l01250"></a><span class="lineno"> 1250</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& input,</div>
|
|
<div class="line"><a id="l01251" name="l01251"></a><span class="lineno"> 1251</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& weight,</div>
|
|
<div class="line"><a id="l01252" name="l01252"></a><span class="lineno"> 1252</span> <span class="keyword">const</span> std::tuple<int, int, int>& stride = {1, 1, 1},</div>
|
|
<div class="line"><a id="l01253" name="l01253"></a><span class="lineno"> 1253</span> <span class="keyword">const</span> std::tuple<int, int, int>& padding = {0, 0, 0},</div>
|
|
<div class="line"><a id="l01254" name="l01254"></a><span class="lineno"> 1254</span> <span class="keyword">const</span> std::tuple<int, int, int>& dilation = {1, 1, 1},</div>
|
|
<div class="line"><a id="l01255" name="l01255"></a><span class="lineno"> 1255</span> <span class="keywordtype">int</span> groups = 1,</div>
|
|
<div class="line"><a id="l01256" name="l01256"></a><span class="lineno"> 1256</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01257" name="l01257"></a><span class="lineno"> 1257</span> </div>
|
|
<div class="line"><a id="l01259" name="l01259"></a><span class="lineno"><a class="line" href="group__ops.html#gaa30bf1adcd78d1c2595d07b215731714"> 1259</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaa30bf1adcd78d1c2595d07b215731714">conv_transpose1d</a>(</div>
|
|
<div class="line"><a id="l01260" name="l01260"></a><span class="lineno"> 1260</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& input,</div>
|
|
<div class="line"><a id="l01261" name="l01261"></a><span class="lineno"> 1261</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& weight,</div>
|
|
<div class="line"><a id="l01262" name="l01262"></a><span class="lineno"> 1262</span> <span class="keywordtype">int</span> stride = 1,</div>
|
|
<div class="line"><a id="l01263" name="l01263"></a><span class="lineno"> 1263</span> <span class="keywordtype">int</span> padding = 0,</div>
|
|
<div class="line"><a id="l01264" name="l01264"></a><span class="lineno"> 1264</span> <span class="keywordtype">int</span> dilation = 1,</div>
|
|
<div class="line"><a id="l01265" name="l01265"></a><span class="lineno"> 1265</span> <span class="keywordtype">int</span> groups = 1,</div>
|
|
<div class="line"><a id="l01266" name="l01266"></a><span class="lineno"> 1266</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01267" name="l01267"></a><span class="lineno"> 1267</span> </div>
|
|
<div class="line"><a id="l01269" name="l01269"></a><span class="lineno"><a class="line" href="group__ops.html#gaebb59971cb9bc45005dc1d398e4f0a3d"> 1269</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaebb59971cb9bc45005dc1d398e4f0a3d">conv_transpose2d</a>(</div>
|
|
<div class="line"><a id="l01270" name="l01270"></a><span class="lineno"> 1270</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& input,</div>
|
|
<div class="line"><a id="l01271" name="l01271"></a><span class="lineno"> 1271</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& weight,</div>
|
|
<div class="line"><a id="l01272" name="l01272"></a><span class="lineno"> 1272</span> <span class="keyword">const</span> std::pair<int, int>& stride = {1, 1},</div>
|
|
<div class="line"><a id="l01273" name="l01273"></a><span class="lineno"> 1273</span> <span class="keyword">const</span> std::pair<int, int>& padding = {0, 0},</div>
|
|
<div class="line"><a id="l01274" name="l01274"></a><span class="lineno"> 1274</span> <span class="keyword">const</span> std::pair<int, int>& dilation = {1, 1},</div>
|
|
<div class="line"><a id="l01275" name="l01275"></a><span class="lineno"> 1275</span> <span class="keywordtype">int</span> groups = 1,</div>
|
|
<div class="line"><a id="l01276" name="l01276"></a><span class="lineno"> 1276</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01277" name="l01277"></a><span class="lineno"> 1277</span> </div>
|
|
<div class="line"><a id="l01279" name="l01279"></a><span class="lineno"><a class="line" href="group__ops.html#ga8db814da631d9cd32a8d6563bf4ac530"> 1279</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga8db814da631d9cd32a8d6563bf4ac530">conv_transpose3d</a>(</div>
|
|
<div class="line"><a id="l01280" name="l01280"></a><span class="lineno"> 1280</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& input,</div>
|
|
<div class="line"><a id="l01281" name="l01281"></a><span class="lineno"> 1281</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& weight,</div>
|
|
<div class="line"><a id="l01282" name="l01282"></a><span class="lineno"> 1282</span> <span class="keyword">const</span> std::tuple<int, int, int>& stride = {1, 1, 1},</div>
|
|
<div class="line"><a id="l01283" name="l01283"></a><span class="lineno"> 1283</span> <span class="keyword">const</span> std::tuple<int, int, int>& padding = {0, 0, 0},</div>
|
|
<div class="line"><a id="l01284" name="l01284"></a><span class="lineno"> 1284</span> <span class="keyword">const</span> std::tuple<int, int, int>& dilation = {1, 1, 1},</div>
|
|
<div class="line"><a id="l01285" name="l01285"></a><span class="lineno"> 1285</span> <span class="keywordtype">int</span> groups = 1,</div>
|
|
<div class="line"><a id="l01286" name="l01286"></a><span class="lineno"> 1286</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01287" name="l01287"></a><span class="lineno"> 1287</span> </div>
|
|
<div class="line"><a id="l01289" name="l01289"></a><span class="lineno"><a class="line" href="group__ops.html#ga6381c7062b232ac21f3fdc3136aaf86e"> 1289</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga6381c7062b232ac21f3fdc3136aaf86e">quantized_matmul</a>(</div>
|
|
<div class="line"><a id="l01290" name="l01290"></a><span class="lineno"> 1290</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& x,</div>
|
|
<div class="line"><a id="l01291" name="l01291"></a><span class="lineno"> 1291</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& w,</div>
|
|
<div class="line"><a id="l01292" name="l01292"></a><span class="lineno"> 1292</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& scales,</div>
|
|
<div class="line"><a id="l01293" name="l01293"></a><span class="lineno"> 1293</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& biases,</div>
|
|
<div class="line"><a id="l01294" name="l01294"></a><span class="lineno"> 1294</span> <span class="keywordtype">bool</span> <a class="code hl_function" href="group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b">transpose</a> = <span class="keyword">true</span>,</div>
|
|
<div class="line"><a id="l01295" name="l01295"></a><span class="lineno"> 1295</span> <span class="keywordtype">int</span> group_size = 64,</div>
|
|
<div class="line"><a id="l01296" name="l01296"></a><span class="lineno"> 1296</span> <span class="keywordtype">int</span> bits = 4,</div>
|
|
<div class="line"><a id="l01297" name="l01297"></a><span class="lineno"> 1297</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01298" name="l01298"></a><span class="lineno"> 1298</span> </div>
|
|
<div class="line"><a id="l01300" name="l01300"></a><span class="lineno"><a class="line" href="group__ops.html#gab43cc28690da7cdd43b43065adbd31da"> 1300</a></span>std::tuple<array, array, array> <a class="code hl_function" href="group__ops.html#gab43cc28690da7cdd43b43065adbd31da">quantize</a>(</div>
|
|
<div class="line"><a id="l01301" name="l01301"></a><span class="lineno"> 1301</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& w,</div>
|
|
<div class="line"><a id="l01302" name="l01302"></a><span class="lineno"> 1302</span> <span class="keywordtype">int</span> group_size = 64,</div>
|
|
<div class="line"><a id="l01303" name="l01303"></a><span class="lineno"> 1303</span> <span class="keywordtype">int</span> bits = 4,</div>
|
|
<div class="line"><a id="l01304" name="l01304"></a><span class="lineno"> 1304</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01305" name="l01305"></a><span class="lineno"> 1305</span> </div>
|
|
<div class="line"><a id="l01307" name="l01307"></a><span class="lineno"><a class="line" href="group__ops.html#gabff758a5c1ce32ad7e8b78aba0164077"> 1307</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gabff758a5c1ce32ad7e8b78aba0164077">dequantize</a>(</div>
|
|
<div class="line"><a id="l01308" name="l01308"></a><span class="lineno"> 1308</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& w,</div>
|
|
<div class="line"><a id="l01309" name="l01309"></a><span class="lineno"> 1309</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& scales,</div>
|
|
<div class="line"><a id="l01310" name="l01310"></a><span class="lineno"> 1310</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& biases,</div>
|
|
<div class="line"><a id="l01311" name="l01311"></a><span class="lineno"> 1311</span> <span class="keywordtype">int</span> group_size = 64,</div>
|
|
<div class="line"><a id="l01312" name="l01312"></a><span class="lineno"> 1312</span> <span class="keywordtype">int</span> bits = 4,</div>
|
|
<div class="line"><a id="l01313" name="l01313"></a><span class="lineno"> 1313</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01314" name="l01314"></a><span class="lineno"> 1314</span> </div>
|
|
<div class="line"><a id="l01316" name="l01316"></a><span class="lineno"><a class="line" href="group__ops.html#ga368a0dc0e5dfb76922e7aa55a95f12f0"> 1316</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga368a0dc0e5dfb76922e7aa55a95f12f0">gather_qmm</a>(</div>
|
|
<div class="line"><a id="l01317" name="l01317"></a><span class="lineno"> 1317</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& x,</div>
|
|
<div class="line"><a id="l01318" name="l01318"></a><span class="lineno"> 1318</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& w,</div>
|
|
<div class="line"><a id="l01319" name="l01319"></a><span class="lineno"> 1319</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& scales,</div>
|
|
<div class="line"><a id="l01320" name="l01320"></a><span class="lineno"> 1320</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& biases,</div>
|
|
<div class="line"><a id="l01321" name="l01321"></a><span class="lineno"> 1321</span> std::optional<array> lhs_indices = std::nullopt,</div>
|
|
<div class="line"><a id="l01322" name="l01322"></a><span class="lineno"> 1322</span> std::optional<array> rhs_indices = std::nullopt,</div>
|
|
<div class="line"><a id="l01323" name="l01323"></a><span class="lineno"> 1323</span> <span class="keywordtype">bool</span> <a class="code hl_function" href="group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b">transpose</a> = <span class="keyword">true</span>,</div>
|
|
<div class="line"><a id="l01324" name="l01324"></a><span class="lineno"> 1324</span> <span class="keywordtype">int</span> group_size = 64,</div>
|
|
<div class="line"><a id="l01325" name="l01325"></a><span class="lineno"> 1325</span> <span class="keywordtype">int</span> bits = 4,</div>
|
|
<div class="line"><a id="l01326" name="l01326"></a><span class="lineno"> 1326</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01327" name="l01327"></a><span class="lineno"> 1327</span> </div>
|
|
<div class="line"><a id="l01329" name="l01329"></a><span class="lineno"><a class="line" href="group__ops.html#gaf5c9735f4690327e1500e04e728fae70"> 1329</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaf5c9735f4690327e1500e04e728fae70">tensordot</a>(</div>
|
|
<div class="line"><a id="l01330" name="l01330"></a><span class="lineno"> 1330</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01331" name="l01331"></a><span class="lineno"> 1331</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b,</div>
|
|
<div class="line"><a id="l01332" name="l01332"></a><span class="lineno"> 1332</span> <span class="keyword">const</span> <span class="keywordtype">int</span> axis = 2,</div>
|
|
<div class="line"><a id="l01333" name="l01333"></a><span class="lineno"> 1333</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01334" name="l01334"></a><span class="lineno"> 1334</span> </div>
|
|
<div class="line"><a id="l01335" name="l01335"></a><span class="lineno"><a class="line" href="group__ops.html#gad7fe00b566f89d607639c1a497cabbc6"> 1335</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaf5c9735f4690327e1500e04e728fae70">tensordot</a>(</div>
|
|
<div class="line"><a id="l01336" name="l01336"></a><span class="lineno"> 1336</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01337" name="l01337"></a><span class="lineno"> 1337</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b,</div>
|
|
<div class="line"><a id="l01338" name="l01338"></a><span class="lineno"> 1338</span> <span class="keyword">const</span> std::vector<int>& axes_a,</div>
|
|
<div class="line"><a id="l01339" name="l01339"></a><span class="lineno"> 1339</span> <span class="keyword">const</span> std::vector<int>& axes_b,</div>
|
|
<div class="line"><a id="l01340" name="l01340"></a><span class="lineno"> 1340</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01341" name="l01341"></a><span class="lineno"> 1341</span> </div>
|
|
<div class="line"><a id="l01343" name="l01343"></a><span class="lineno"><a class="line" href="group__ops.html#ga866af24e10db2797e1c5a5986dbf6c0d"> 1343</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga866af24e10db2797e1c5a5986dbf6c0d">outer</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01344" name="l01344"></a><span class="lineno"> 1344</span> </div>
|
|
<div class="line"><a id="l01346" name="l01346"></a><span class="lineno"><a class="line" href="group__ops.html#ga654fec16a9746b390916697a2ab2546e"> 1346</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga654fec16a9746b390916697a2ab2546e">inner</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01347" name="l01347"></a><span class="lineno"> 1347</span> </div>
|
|
<div class="line"><a id="l01349" name="l01349"></a><span class="lineno"><a class="line" href="group__ops.html#ga82a53e083205a965387b3c3e2463244a"> 1349</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga82a53e083205a965387b3c3e2463244a">addmm</a>(</div>
|
|
<div class="line"><a id="l01350" name="l01350"></a><span class="lineno"> 1350</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> c,</div>
|
|
<div class="line"><a id="l01351" name="l01351"></a><span class="lineno"> 1351</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> a,</div>
|
|
<div class="line"><a id="l01352" name="l01352"></a><span class="lineno"> 1352</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> b,</div>
|
|
<div class="line"><a id="l01353" name="l01353"></a><span class="lineno"> 1353</span> <span class="keyword">const</span> <span class="keywordtype">float</span>& alpha = 1.f,</div>
|
|
<div class="line"><a id="l01354" name="l01354"></a><span class="lineno"> 1354</span> <span class="keyword">const</span> <span class="keywordtype">float</span>& beta = 1.f,</div>
|
|
<div class="line"><a id="l01355" name="l01355"></a><span class="lineno"> 1355</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01356" name="l01356"></a><span class="lineno"> 1356</span> </div>
|
|
<div class="line"><a id="l01358" name="l01358"></a><span class="lineno"><a class="line" href="group__ops.html#ga6b76c8ea46b19e6866af155fa5910be6"> 1358</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga6b76c8ea46b19e6866af155fa5910be6">block_masked_mm</a>(</div>
|
|
<div class="line"><a id="l01359" name="l01359"></a><span class="lineno"> 1359</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> a,</div>
|
|
<div class="line"><a id="l01360" name="l01360"></a><span class="lineno"> 1360</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> b,</div>
|
|
<div class="line"><a id="l01361" name="l01361"></a><span class="lineno"> 1361</span> <span class="keywordtype">int</span> block_size,</div>
|
|
<div class="line"><a id="l01362" name="l01362"></a><span class="lineno"> 1362</span> std::optional<array> mask_out = std::nullopt,</div>
|
|
<div class="line"><a id="l01363" name="l01363"></a><span class="lineno"> 1363</span> std::optional<array> mask_lhs = std::nullopt,</div>
|
|
<div class="line"><a id="l01364" name="l01364"></a><span class="lineno"> 1364</span> std::optional<array> mask_rhs = std::nullopt,</div>
|
|
<div class="line"><a id="l01365" name="l01365"></a><span class="lineno"> 1365</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01366" name="l01366"></a><span class="lineno"> 1366</span> </div>
|
|
<div class="line"><a id="l01368" name="l01368"></a><span class="lineno"><a class="line" href="group__ops.html#ga8d50480266d258cac40ff51bcb0fc6a7"> 1368</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga8d50480266d258cac40ff51bcb0fc6a7">gather_mm</a>(</div>
|
|
<div class="line"><a id="l01369" name="l01369"></a><span class="lineno"> 1369</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> a,</div>
|
|
<div class="line"><a id="l01370" name="l01370"></a><span class="lineno"> 1370</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> b,</div>
|
|
<div class="line"><a id="l01371" name="l01371"></a><span class="lineno"> 1371</span> std::optional<array> lhs_indices = std::nullopt,</div>
|
|
<div class="line"><a id="l01372" name="l01372"></a><span class="lineno"> 1372</span> std::optional<array> rhs_indices = std::nullopt,</div>
|
|
<div class="line"><a id="l01373" name="l01373"></a><span class="lineno"> 1373</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01374" name="l01374"></a><span class="lineno"> 1374</span> </div>
|
|
<div class="line"><a id="l01376" name="l01376"></a><span class="lineno"><a class="line" href="group__ops.html#ga9236b085a88ead3128ed8079d009cac6"> 1376</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga9236b085a88ead3128ed8079d009cac6">diagonal</a>(</div>
|
|
<div class="line"><a id="l01377" name="l01377"></a><span class="lineno"> 1377</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01378" name="l01378"></a><span class="lineno"> 1378</span> <span class="keywordtype">int</span> offset = 0,</div>
|
|
<div class="line"><a id="l01379" name="l01379"></a><span class="lineno"> 1379</span> <span class="keywordtype">int</span> axis1 = 0,</div>
|
|
<div class="line"><a id="l01380" name="l01380"></a><span class="lineno"> 1380</span> <span class="keywordtype">int</span> axis2 = 1,</div>
|
|
<div class="line"><a id="l01381" name="l01381"></a><span class="lineno"> 1381</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01382" name="l01382"></a><span class="lineno"> 1382</span> </div>
|
|
<div class="line"><a id="l01384" name="l01384"></a><span class="lineno"><a class="line" href="group__ops.html#ga11af511875640e1fa88e0ca87e199344"> 1384</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga11af511875640e1fa88e0ca87e199344">diag</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> k = 0, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01385" name="l01385"></a><span class="lineno"> 1385</span> </div>
|
|
<div class="line"><a id="l01387" name="l01387"></a><span class="lineno"><a class="line" href="group__ops.html#gabf786129c7660ed8d5acb5499bc6fefd"> 1387</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gabf786129c7660ed8d5acb5499bc6fefd">trace</a>(</div>
|
|
<div class="line"><a id="l01388" name="l01388"></a><span class="lineno"> 1388</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01389" name="l01389"></a><span class="lineno"> 1389</span> <span class="keywordtype">int</span> offset,</div>
|
|
<div class="line"><a id="l01390" name="l01390"></a><span class="lineno"> 1390</span> <span class="keywordtype">int</span> axis1,</div>
|
|
<div class="line"><a id="l01391" name="l01391"></a><span class="lineno"> 1391</span> <span class="keywordtype">int</span> axis2,</div>
|
|
<div class="line"><a id="l01392" name="l01392"></a><span class="lineno"> 1392</span> <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype,</div>
|
|
<div class="line"><a id="l01393" name="l01393"></a><span class="lineno"> 1393</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01394" name="l01394"></a><span class="lineno"><a class="line" href="group__ops.html#ga5ed43c2dbf7d6cbddbaa2fd682deaafd"> 1394</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gabf786129c7660ed8d5acb5499bc6fefd">trace</a>(</div>
|
|
<div class="line"><a id="l01395" name="l01395"></a><span class="lineno"> 1395</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01396" name="l01396"></a><span class="lineno"> 1396</span> <span class="keywordtype">int</span> offset,</div>
|
|
<div class="line"><a id="l01397" name="l01397"></a><span class="lineno"> 1397</span> <span class="keywordtype">int</span> axis1,</div>
|
|
<div class="line"><a id="l01398" name="l01398"></a><span class="lineno"> 1398</span> <span class="keywordtype">int</span> axis2,</div>
|
|
<div class="line"><a id="l01399" name="l01399"></a><span class="lineno"> 1399</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01400" name="l01400"></a><span class="lineno"><a class="line" href="group__ops.html#gaf25c00108feaafaa6350a4434cb0062e"> 1400</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gabf786129c7660ed8d5acb5499bc6fefd">trace</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01401" name="l01401"></a><span class="lineno"> 1401</span> </div>
|
|
<div class="line"><a id="l01407" name="l01407"></a><span class="lineno"><a class="line" href="group__ops.html#gac4a51a68fbe1725436b026d2fbb95759"> 1407</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#gac4a51a68fbe1725436b026d2fbb95759">depends</a>(</div>
|
|
<div class="line"><a id="l01408" name="l01408"></a><span class="lineno"> 1408</span> <span class="keyword">const</span> std::vector<array>& inputs,</div>
|
|
<div class="line"><a id="l01409" name="l01409"></a><span class="lineno"> 1409</span> <span class="keyword">const</span> std::vector<array>& dependencies);</div>
|
|
<div class="line"><a id="l01410" name="l01410"></a><span class="lineno"> 1410</span> </div>
|
|
<div class="line"><a id="l01412" name="l01412"></a><span class="lineno"><a class="line" href="group__ops.html#gaba4d25e7a2bf87ba4feb7837ec7fa396"> 1412</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaba4d25e7a2bf87ba4feb7837ec7fa396">atleast_1d</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01413" name="l01413"></a><span class="lineno"><a class="line" href="group__ops.html#ga08ca172ce80157c916c89dd0b45b95c5"> 1413</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#gaba4d25e7a2bf87ba4feb7837ec7fa396">atleast_1d</a>(</div>
|
|
<div class="line"><a id="l01414" name="l01414"></a><span class="lineno"> 1414</span> <span class="keyword">const</span> std::vector<array>& a,</div>
|
|
<div class="line"><a id="l01415" name="l01415"></a><span class="lineno"> 1415</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01416" name="l01416"></a><span class="lineno"><a class="line" href="group__ops.html#gaeeb7f5bb88aa32a3ac2be1f39c5f8087"> 1416</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaeeb7f5bb88aa32a3ac2be1f39c5f8087">atleast_2d</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01417" name="l01417"></a><span class="lineno"><a class="line" href="group__ops.html#ga9950299a80c2562f13448758f856d1f5"> 1417</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#gaeeb7f5bb88aa32a3ac2be1f39c5f8087">atleast_2d</a>(</div>
|
|
<div class="line"><a id="l01418" name="l01418"></a><span class="lineno"> 1418</span> <span class="keyword">const</span> std::vector<array>& a,</div>
|
|
<div class="line"><a id="l01419" name="l01419"></a><span class="lineno"> 1419</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01420" name="l01420"></a><span class="lineno"><a class="line" href="group__ops.html#ga4afd919601e67782ff964465919956a0"> 1420</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga4afd919601e67782ff964465919956a0">atleast_3d</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01421" name="l01421"></a><span class="lineno"><a class="line" href="group__ops.html#gaffdf742ad79440a60dda40062a8074fe"> 1421</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#ga4afd919601e67782ff964465919956a0">atleast_3d</a>(</div>
|
|
<div class="line"><a id="l01422" name="l01422"></a><span class="lineno"> 1422</span> <span class="keyword">const</span> std::vector<array>& a,</div>
|
|
<div class="line"><a id="l01423" name="l01423"></a><span class="lineno"> 1423</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01424" name="l01424"></a><span class="lineno"> 1424</span> </div>
|
|
<div class="line"><a id="l01429" name="l01429"></a><span class="lineno"><a class="line" href="group__ops.html#ga6d5f5f72362488b956cdc4615ef6c636"> 1429</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga6d5f5f72362488b956cdc4615ef6c636">number_of_elements</a>(</div>
|
|
<div class="line"><a id="l01430" name="l01430"></a><span class="lineno"> 1430</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01431" name="l01431"></a><span class="lineno"> 1431</span> std::vector<int> axes,</div>
|
|
<div class="line"><a id="l01432" name="l01432"></a><span class="lineno"> 1432</span> <span class="keywordtype">bool</span> inverted,</div>
|
|
<div class="line"><a id="l01433" name="l01433"></a><span class="lineno"> 1433</span> <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype = <a class="code hl_variable" href="namespacemlx_1_1core.html#a5d6373aad1444edc9de1eb07bfe5cad3">int32</a>,</div>
|
|
<div class="line"><a id="l01434" name="l01434"></a><span class="lineno"> 1434</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01435" name="l01435"></a><span class="lineno"> 1435</span> </div>
|
|
<div class="line"><a id="l01436" name="l01436"></a><span class="lineno"><a class="line" href="group__ops.html#ga5b596906bf8cdc8d97ed6ddc9aeb4c23"> 1436</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga5b596906bf8cdc8d97ed6ddc9aeb4c23">conjugate</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01437" name="l01437"></a><span class="lineno"> 1437</span> </div>
|
|
<div class="line"><a id="l01439" name="l01439"></a><span class="lineno"><a class="line" href="group__ops.html#ga752fd2707dabb05d0308ba3d55346ada"> 1439</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga752fd2707dabb05d0308ba3d55346ada">bitwise_and</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01440" name="l01440"></a><span class="lineno"><a class="line" href="group__ops.html#gaf0d232de4cbfffda1e2c838f8afdf6ff"> 1440</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaf0d232de4cbfffda1e2c838f8afdf6ff">operator&</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
|
<div class="line"><a id="l01441" name="l01441"></a><span class="lineno"> 1441</span> </div>
|
|
<div class="line"><a id="l01443" name="l01443"></a><span class="lineno"><a class="line" href="group__ops.html#ga8af4f22c08c11c4ffab7e3d45e0f3cd6"> 1443</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga8af4f22c08c11c4ffab7e3d45e0f3cd6">bitwise_or</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01444" name="l01444"></a><span class="lineno"><a class="line" href="group__ops.html#ga52392a2a98f09a80da8d338c4908bd02"> 1444</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga52392a2a98f09a80da8d338c4908bd02">operator|</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
|
<div class="line"><a id="l01445" name="l01445"></a><span class="lineno"> 1445</span> </div>
|
|
<div class="line"><a id="l01447" name="l01447"></a><span class="lineno"><a class="line" href="group__ops.html#ga3188638fba3a60e264baf69956a1e08b"> 1447</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga3188638fba3a60e264baf69956a1e08b">bitwise_xor</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01448" name="l01448"></a><span class="lineno"><a class="line" href="group__ops.html#gac3a6fe18694e84b3d63458e9553ac181"> 1448</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac3a6fe18694e84b3d63458e9553ac181">operator^</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
|
<div class="line"><a id="l01449" name="l01449"></a><span class="lineno"> 1449</span> </div>
|
|
<div class="line"><a id="l01451" name="l01451"></a><span class="lineno"><a class="line" href="group__ops.html#ga89682bf78491761e062d4ee7bef0c829"> 1451</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga89682bf78491761e062d4ee7bef0c829">left_shift</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01452" name="l01452"></a><span class="lineno"><a class="line" href="group__ops.html#gad656c30f9fd7d9467e405657b325aa7e"> 1452</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gad656c30f9fd7d9467e405657b325aa7e">operator<<</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
|
<div class="line"><a id="l01453" name="l01453"></a><span class="lineno"> 1453</span> </div>
|
|
<div class="line"><a id="l01455" name="l01455"></a><span class="lineno"><a class="line" href="group__ops.html#gafa376ad57d38ba87378f0272dc379b23"> 1455</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gafa376ad57d38ba87378f0272dc379b23">right_shift</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01456" name="l01456"></a><span class="lineno"><a class="line" href="group__ops.html#ga498b61f7e8f056ae00297fa0dc17303a"> 1456</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga498b61f7e8f056ae00297fa0dc17303a">operator>></a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
|
<div class="line"><a id="l01457" name="l01457"></a><span class="lineno"> 1457</span> </div>
|
|
<div class="line"><a id="l01458" name="l01458"></a><span class="lineno"><a class="line" href="group__ops.html#ga3602aa91b7b124a0b41ec1b2137a1b02"> 1458</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga3602aa91b7b124a0b41ec1b2137a1b02">view</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a>& dtype, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01459" name="l01459"></a><span class="lineno"> 1459</span> </div>
|
|
<div class="line"><a id="l01461" name="l01461"></a><span class="lineno"><a class="line" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c"> 1461</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c">roll</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> shift, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01462" name="l01462"></a><span class="lineno"><a class="line" href="group__ops.html#gab30fb942bb6ee00bb50da04033f949a9"> 1462</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c">roll</a>(</div>
|
|
<div class="line"><a id="l01463" name="l01463"></a><span class="lineno"> 1463</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01464" name="l01464"></a><span class="lineno"> 1464</span> <span class="keyword">const</span> std::vector<int>& shift,</div>
|
|
<div class="line"><a id="l01465" name="l01465"></a><span class="lineno"> 1465</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01466" name="l01466"></a><span class="lineno"><a class="line" href="group__ops.html#ga8694ec137165752cb6d8a36a6b7c3436"> 1466</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c">roll</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> shift, <span class="keywordtype">int</span> axis, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01467" name="l01467"></a><span class="lineno"><a class="line" href="group__ops.html#ga665f502ecc96f1f4467556b784abf9ae"> 1467</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c">roll</a>(</div>
|
|
<div class="line"><a id="l01468" name="l01468"></a><span class="lineno"> 1468</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01469" name="l01469"></a><span class="lineno"> 1469</span> <span class="keywordtype">int</span> shift,</div>
|
|
<div class="line"><a id="l01470" name="l01470"></a><span class="lineno"> 1470</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l01471" name="l01471"></a><span class="lineno"> 1471</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01472" name="l01472"></a><span class="lineno"><a class="line" href="group__ops.html#ga4c335f86ff2a05daf8dd8aedc628aa59"> 1472</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c">roll</a>(</div>
|
|
<div class="line"><a id="l01473" name="l01473"></a><span class="lineno"> 1473</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01474" name="l01474"></a><span class="lineno"> 1474</span> <span class="keyword">const</span> std::vector<int>& shift,</div>
|
|
<div class="line"><a id="l01475" name="l01475"></a><span class="lineno"> 1475</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l01476" name="l01476"></a><span class="lineno"> 1476</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01477" name="l01477"></a><span class="lineno"><a class="line" href="group__ops.html#ga6c6fe942a8f7c573a1cb2342566f74d2"> 1477</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c">roll</a>(</div>
|
|
<div class="line"><a id="l01478" name="l01478"></a><span class="lineno"> 1478</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
|
<div class="line"><a id="l01479" name="l01479"></a><span class="lineno"> 1479</span> <span class="keyword">const</span> std::vector<int>& shift,</div>
|
|
<div class="line"><a id="l01480" name="l01480"></a><span class="lineno"> 1480</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l01481" name="l01481"></a><span class="lineno"> 1481</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01482" name="l01482"></a><span class="lineno"> 1482</span> </div>
|
|
<div class="line"><a id="l01483" name="l01483"></a><span class="lineno"> 1483</span><span class="comment">/* The real part of a complex array. */</span></div>
|
|
<div class="line"><a id="l01484" name="l01484"></a><span class="lineno"><a class="line" href="group__ops.html#gaf8913cabeb9fb193ba687aaeb2087764"> 1484</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaf8913cabeb9fb193ba687aaeb2087764">real</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01485" name="l01485"></a><span class="lineno"> 1485</span> </div>
|
|
<div class="line"><a id="l01486" name="l01486"></a><span class="lineno"> 1486</span><span class="comment">/* The imaginary part of a complex array. */</span></div>
|
|
<div class="line"><a id="l01487" name="l01487"></a><span class="lineno"><a class="line" href="group__ops.html#ga7ff592a64d528f0cf4f3d098465da029"> 1487</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7ff592a64d528f0cf4f3d098465da029">imag</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01488" name="l01488"></a><span class="lineno"> 1488</span> </div>
|
|
<div class="line"><a id="l01491" name="l01491"></a><span class="lineno"> 1491</span>} <span class="comment">// namespace mlx::core</span></div>
|
|
<div class="ttc" id="aarray_8h_html"><div class="ttname"><a href="array_8h.html">array.h</a></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:20</div></div>
|
|
<div class="ttc" id="adevice_8h_html"><div class="ttname"><a href="device_8h.html">device.h</a></div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga05881a4157cd113c9392d168a79e6673"><div class="ttname"><a href="group__ops.html#ga05881a4157cd113c9392d168a79e6673">mlx::core::scatter_max</a></div><div class="ttdeci">array scatter_max(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})</div><div class="ttdoc">Scatter and max updates to given linear indices.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga05b4c6054d028107869511f927da01cd"><div class="ttname"><a href="group__ops.html#ga05b4c6054d028107869511f927da01cd">mlx::core::floor_divide</a></div><div class="ttdeci">array floor_divide(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Compute integer division.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga07aa8059adba5b9a8818027b8aafd31e"><div class="ttname"><a href="group__ops.html#ga07aa8059adba5b9a8818027b8aafd31e">mlx::core::radians</a></div><div class="ttdeci">array radians(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Convert the elements of an array from Degrees to Radians.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga08bec7cb10c84466487b507fc5bf9776"><div class="ttname"><a href="group__ops.html#ga08bec7cb10c84466487b507fc5bf9776">mlx::core::arccos</a></div><div class="ttdeci">array arccos(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Arc Cosine of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga0ca16b7579dfc899f3f7fd40245ba7c5"><div class="ttname"><a href="group__ops.html#ga0ca16b7579dfc899f3f7fd40245ba7c5">mlx::core::scatter_min</a></div><div class="ttdeci">array scatter_min(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})</div><div class="ttdoc">Scatter and min updates to given linear indices.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga0d49e0c7011d0573c369c13c8f045a09"><div class="ttname"><a href="group__ops.html#ga0d49e0c7011d0573c369c13c8f045a09">mlx::core::less_equal</a></div><div class="ttdeci">array less_equal(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Returns bool array with (a <= b) element-wise.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga0d71dfbc14ef3ed564b0c5ee26af680f"><div class="ttname"><a href="group__ops.html#ga0d71dfbc14ef3ed564b0c5ee26af680f">mlx::core::cumprod</a></div><div class="ttdeci">array cumprod(const array &a, int axis, bool reverse=false, bool inclusive=true, StreamOrDevice s={})</div><div class="ttdoc">Cumulative product of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga0e58c24fc5668e5a521e5b45e8370a62"><div class="ttname"><a href="group__ops.html#ga0e58c24fc5668e5a521e5b45e8370a62">mlx::core::astype</a></div><div class="ttdeci">array astype(array a, Dtype dtype, StreamOrDevice s={})</div><div class="ttdoc">Convert an array to the given data type.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga102f23aa0b0c3d3296a321c694617aa1"><div class="ttname"><a href="group__ops.html#ga102f23aa0b0c3d3296a321c694617aa1">mlx::core::rsqrt</a></div><div class="ttdeci">array rsqrt(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Square root and reciprocal the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga11af511875640e1fa88e0ca87e199344"><div class="ttname"><a href="group__ops.html#ga11af511875640e1fa88e0ca87e199344">mlx::core::diag</a></div><div class="ttdeci">array diag(const array &a, int k=0, StreamOrDevice s={})</div><div class="ttdoc">Extract diagonal from a 2d array or create a diagonal matrix.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga1234e4c39cfa79f19d4bdb5b8ea4d45e"><div class="ttname"><a href="group__ops.html#ga1234e4c39cfa79f19d4bdb5b8ea4d45e">mlx::core::square</a></div><div class="ttdeci">array square(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Square the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga1404ecceff83fd9b9139b7520f55e096"><div class="ttname"><a href="group__ops.html#ga1404ecceff83fd9b9139b7520f55e096">mlx::core::ceil</a></div><div class="ttdeci">array ceil(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Ceil the element of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga144228d7222d15af3a135b8b0f3fa21b"><div class="ttname"><a href="group__ops.html#ga144228d7222d15af3a135b8b0f3fa21b">mlx::core::log2</a></div><div class="ttdeci">array log2(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Log base 2 of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga157cd7c23f9b306fee2e1eb2b9bf1dd8"><div class="ttname"><a href="group__ops.html#ga157cd7c23f9b306fee2e1eb2b9bf1dd8">mlx::core::clip</a></div><div class="ttdeci">array clip(const array &a, const std::optional< array > &a_min=std::nullopt, const std::optional< array > &a_max=std::nullopt, StreamOrDevice s={})</div><div class="ttdoc">Clip (limit) the values in an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga175592792471b0ffb45196dca4711ba6"><div class="ttname"><a href="group__ops.html#ga175592792471b0ffb45196dca4711ba6">mlx::core::isnan</a></div><div class="ttdeci">array isnan(const array &a, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga1940523da381ed7be50656a3bc465ff3"><div class="ttname"><a href="group__ops.html#ga1940523da381ed7be50656a3bc465ff3">mlx::core::isneginf</a></div><div class="ttdeci">array isneginf(const array &a, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga196c240d3d0fcbb4713802c485e15133"><div class="ttname"><a href="group__ops.html#ga196c240d3d0fcbb4713802c485e15133">mlx::core::subtract</a></div><div class="ttdeci">array subtract(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Subtract two arrays.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga19c1bf6929fe8d66b9cd408946aea6a8"><div class="ttname"><a href="group__ops.html#ga19c1bf6929fe8d66b9cd408946aea6a8">mlx::core::cummin</a></div><div class="ttdeci">array cummin(const array &a, int axis, bool reverse=false, bool inclusive=true, StreamOrDevice s={})</div><div class="ttdoc">Cumulative min of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga1fdcc7fc8819caf2e6f1c327ed4e9b9e"><div class="ttname"><a href="group__ops.html#ga1fdcc7fc8819caf2e6f1c327ed4e9b9e">mlx::core::log10</a></div><div class="ttdeci">array log10(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Log base 10 of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga20a1f4270c35b0fa544f5105a87a1604"><div class="ttname"><a href="group__ops.html#ga20a1f4270c35b0fa544f5105a87a1604">mlx::core::log1p</a></div><div class="ttdeci">array log1p(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Natural logarithm of one plus elements in the array: log(1 + a).</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga20f1a1a8c0cd6206485f9363f3915faa"><div class="ttname"><a href="group__ops.html#ga20f1a1a8c0cd6206485f9363f3915faa">mlx::core::sign</a></div><div class="ttdeci">array sign(const array &a, StreamOrDevice s={})</div><div class="ttdoc">The sign of the elements in an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga2181b71cda88007a3092be4795ff0715"><div class="ttname"><a href="group__ops.html#ga2181b71cda88007a3092be4795ff0715">mlx::core::cosh</a></div><div class="ttdeci">array cosh(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Hyperbolic Cosine of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga2236e5dfc7e52e28abf6c21675d0a51e"><div class="ttname"><a href="group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e">mlx::core::conv_general</a></div><div class="ttdeci">array conv_general(array input, array weight, std::vector< int > stride={}, std::vector< int > padding_lo={}, std::vector< int > padding_hi={}, std::vector< int > kernel_dilation={}, std::vector< int > input_dilation={}, int groups=1, bool flip=false, StreamOrDevice s={})</div><div class="ttdoc">General convolution with a filter.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga23768728e4dd070c917fbb0ed0d0c2ec"><div class="ttname"><a href="group__ops.html#ga23768728e4dd070c917fbb0ed0d0c2ec">mlx::core::logical_or</a></div><div class="ttdeci">array logical_or(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Logical or of two arrays.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga24067d10a842db2c9d509ea48135a2c3"><div class="ttname"><a href="group__ops.html#ga24067d10a842db2c9d509ea48135a2c3">mlx::core::moveaxis</a></div><div class="ttdeci">array moveaxis(const array &a, int source, int destination, StreamOrDevice s={})</div><div class="ttdoc">Move an axis of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga26c33f5cdb6fc10d272acd6e208034e0"><div class="ttname"><a href="group__ops.html#ga26c33f5cdb6fc10d272acd6e208034e0">mlx::core::operator*</a></div><div class="ttdeci">array operator*(const array &a, const array &b)</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga26e5a043eaaaf066d1400adac9c11d0c"><div class="ttname"><a href="group__ops.html#ga26e5a043eaaaf066d1400adac9c11d0c">mlx::core::operator+</a></div><div class="ttdeci">array operator+(const array &a, const array &b)</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga27af56a98270d4d76d139f0f9171b83a"><div class="ttname"><a href="group__ops.html#ga27af56a98270d4d76d139f0f9171b83a">mlx::core::operator||</a></div><div class="ttdeci">array operator||(const array &a, const array &b)</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga28f22c5d2c399eee53be7b3facc11103"><div class="ttname"><a href="group__ops.html#ga28f22c5d2c399eee53be7b3facc11103">mlx::core::not_equal</a></div><div class="ttdeci">array not_equal(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Returns the bool array with (a != b) element-wise.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga292a335240fd5d6d625fb7a340ff5eb0"><div class="ttname"><a href="group__ops.html#ga292a335240fd5d6d625fb7a340ff5eb0">mlx::core::erf</a></div><div class="ttdeci">array erf(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Computes the error function of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga297f853b3d90ec8ae81263977ba2ddb1"><div class="ttname"><a href="group__ops.html#ga297f853b3d90ec8ae81263977ba2ddb1">mlx::core::sqrt</a></div><div class="ttdeci">array sqrt(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Square root the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga2a466024f8061febc0a64be557644cb0"><div class="ttname"><a href="group__ops.html#ga2a466024f8061febc0a64be557644cb0">mlx::core::std</a></div><div class="ttdeci">array std(const array &a, bool keepdims, int ddof=0, StreamOrDevice s={})</div><div class="ttdoc">Computes the standard deviation of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga2d32d67cfd76785a72c43d89b94dc7d7"><div class="ttname"><a href="group__ops.html#ga2d32d67cfd76785a72c43d89b94dc7d7">mlx::core::add</a></div><div class="ttdeci">array add(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Add two arrays.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga2d74d43f007a069384e89d8416525331"><div class="ttname"><a href="group__ops.html#ga2d74d43f007a069384e89d8416525331">mlx::core::round</a></div><div class="ttdeci">array round(const array &a, int decimals, StreamOrDevice s={})</div><div class="ttdoc">Round a floating point number.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga30d47e08093c03a3676f235f9f559411"><div class="ttname"><a href="group__ops.html#ga30d47e08093c03a3676f235f9f559411">mlx::core::conv1d</a></div><div class="ttdeci">array conv1d(const array &input, const array &weight, int stride=1, int padding=0, int dilation=1, int groups=1, StreamOrDevice s={})</div><div class="ttdoc">1D convolution with a filter</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga3188638fba3a60e264baf69956a1e08b"><div class="ttname"><a href="group__ops.html#ga3188638fba3a60e264baf69956a1e08b">mlx::core::bitwise_xor</a></div><div class="ttdeci">array bitwise_xor(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Bitwise exclusive or.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga33638dc3a9972dd02be12d0eb85f9bde"><div class="ttname"><a href="group__ops.html#ga33638dc3a9972dd02be12d0eb85f9bde">mlx::core::equal</a></div><div class="ttdeci">array equal(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Returns the bool array with (a == b) element-wise.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga34573eb2ead179839579142f3aedb3b8"><div class="ttname"><a href="group__ops.html#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_ga3602aa91b7b124a0b41ec1b2137a1b02"><div class="ttname"><a href="group__ops.html#ga3602aa91b7b124a0b41ec1b2137a1b02">mlx::core::view</a></div><div class="ttdeci">array view(const array &a, const Dtype &dtype, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga368a0dc0e5dfb76922e7aa55a95f12f0"><div class="ttname"><a href="group__ops.html#ga368a0dc0e5dfb76922e7aa55a95f12f0">mlx::core::gather_qmm</a></div><div class="ttdeci">array gather_qmm(const array &x, const array &w, const array &scales, const array &biases, std::optional< array > lhs_indices=std::nullopt, std::optional< array > rhs_indices=std::nullopt, bool transpose=true, int group_size=64, int bits=4, StreamOrDevice s={})</div><div class="ttdoc">Compute matrix products with matrix-level gather.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga36bc28f1deb2fe668ca9ae1e447b6b1f"><div class="ttname"><a href="group__ops.html#ga36bc28f1deb2fe668ca9ae1e447b6b1f">mlx::core::stop_gradient</a></div><div class="ttdeci">array stop_gradient(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Stop the flow of gradients.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga3708b5bcb61e2c63d213c4ce6ad0ffc0"><div class="ttname"><a href="group__ops.html#ga3708b5bcb61e2c63d213c4ce6ad0ffc0">mlx::core::scatter_prod</a></div><div class="ttdeci">array scatter_prod(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})</div><div class="ttdoc">Scatter and prod updates to given indices.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga3875660e4ce2c8add8bfcf8144078708"><div class="ttname"><a href="group__ops.html#ga3875660e4ce2c8add8bfcf8144078708">mlx::core::slice_update</a></div><div class="ttdeci">array slice_update(const array &src, const array &update, std::vector< int > start, std::vector< int > stop, std::vector< int > strides, StreamOrDevice s={})</div><div class="ttdoc">Update a slice from the source array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga39dfdf72b556012aa35ff27a94116e74"><div class="ttname"><a href="group__ops.html#ga39dfdf72b556012aa35ff27a94116e74">mlx::core::cos</a></div><div class="ttdeci">array cos(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Cosine of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga3a41895f25ed083a36994d95fa102546"><div class="ttname"><a href="group__ops.html#ga3a41895f25ed083a36994d95fa102546">mlx::core::operator>=</a></div><div class="ttdeci">array operator>=(const array &a, const array &b)</div><div class="ttdef"><b>Definition</b> ops.h:345</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga3a70569b50e1083c5ded199d73fb960c"><div class="ttname"><a href="group__ops.html#ga3a70569b50e1083c5ded199d73fb960c">mlx::core::degrees</a></div><div class="ttdeci">array degrees(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Convert the elements of an array from Radians to Degrees.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga3b1b90ef1275ca17655b6d7f25d3ee68"><div class="ttname"><a href="group__ops.html#ga3b1b90ef1275ca17655b6d7f25d3ee68">mlx::core::all</a></div><div class="ttdeci">array all(const array &a, bool keepdims, StreamOrDevice s={})</div><div class="ttdoc">True if all elements in the array are true (or non-zero).</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga3f10e89a4bcb1a8fa44fb33b8d1176a5"><div class="ttname"><a href="group__ops.html#ga3f10e89a4bcb1a8fa44fb33b8d1176a5">mlx::core::tan</a></div><div class="ttdeci">array tan(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Tangent of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga45e9e68246b0d1cf03c3cc9c9e7e6ae3"><div class="ttname"><a href="group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3">mlx::core::eye</a></div><div class="ttdeci">array eye(int n, int m, int k, Dtype dtype, StreamOrDevice s={})</div><div class="ttdoc">Fill an array of the given shape (n,m) with ones in the specified diagonal k, and zeros everywhere el...</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga484eaa10d5e19a4ca46d3a9cd9fab600"><div class="ttname"><a href="group__ops.html#ga484eaa10d5e19a4ca46d3a9cd9fab600">mlx::core::identity</a></div><div class="ttdeci">array identity(int n, Dtype dtype, StreamOrDevice s={})</div><div class="ttdoc">Create a square matrix of shape (n,n) of zeros, and ones in the major diagonal.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga498b61f7e8f056ae00297fa0dc17303a"><div class="ttname"><a href="group__ops.html#ga498b61f7e8f056ae00297fa0dc17303a">mlx::core::operator>></a></div><div class="ttdeci">array operator>>(const array &a, const array &b)</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga49ba00c090f81f331c91b0c97040bce0"><div class="ttname"><a href="group__ops.html#ga49ba00c090f81f331c91b0c97040bce0">mlx::core::minimum</a></div><div class="ttdeci">array minimum(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Element-wise minimum between two arrays.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga4a09b7241d564d92548bc2773eb1d544"><div class="ttname"><a href="group__ops.html#ga4a09b7241d564d92548bc2773eb1d544">mlx::core::prod</a></div><div class="ttdeci">array prod(const array &a, bool keepdims, StreamOrDevice s={})</div><div class="ttdoc">The product of all elements of the array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga4afd919601e67782ff964465919956a0"><div class="ttname"><a href="group__ops.html#ga4afd919601e67782ff964465919956a0">mlx::core::atleast_3d</a></div><div class="ttdeci">array atleast_3d(const array &a, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga4c8b8a1632944acaae50f0de6c23ece6"><div class="ttname"><a href="group__ops.html#ga4c8b8a1632944acaae50f0de6c23ece6">mlx::core::operator<=</a></div><div class="ttdeci">array operator<=(const array &a, const array &b)</div><div class="ttdef"><b>Definition</b> ops.h:373</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga4d29556bb93e2f66916116cf1f062b36"><div class="ttname"><a href="group__ops.html#ga4d29556bb93e2f66916116cf1f062b36">mlx::core::reciprocal</a></div><div class="ttdeci">array reciprocal(const array &a, StreamOrDevice s={})</div><div class="ttdoc">The reciprocal (1/x) of the elements in an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga4f3389e5b89e70e862e7d2b40d6c7f78"><div class="ttname"><a href="group__ops.html#ga4f3389e5b89e70e862e7d2b40d6c7f78">mlx::core::tri</a></div><div class="ttdeci">array tri(int n, int m, int k, Dtype type, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga50aa98754b412bb57c083f6e3e95061f"><div class="ttname"><a href="group__ops.html#ga50aa98754b412bb57c083f6e3e95061f">mlx::core::flatten</a></div><div class="ttdeci">array flatten(const array &a, int start_axis, int end_axis=-1, StreamOrDevice s={})</div><div class="ttdoc">Flatten the dimensions in the range [start_axis, end_axis] .</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga51eac95c04400921c54716de14b52491"><div class="ttname"><a href="group__ops.html#ga51eac95c04400921c54716de14b52491">mlx::core::isclose</a></div><div class="ttdeci">array isclose(const array &a, const array &b, double rtol=1e-5, double atol=1e-8, bool equal_nan=false, StreamOrDevice s={})</div><div class="ttdoc">Returns a boolean array where two arrays are element-wise equal within the specified tolerance.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga52392a2a98f09a80da8d338c4908bd02"><div class="ttname"><a href="group__ops.html#ga52392a2a98f09a80da8d338c4908bd02">mlx::core::operator|</a></div><div class="ttdeci">array operator|(const array &a, const array &b)</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga5487dd887c43e5341f3e68ffe47f0f5a"><div class="ttname"><a href="group__ops.html#ga5487dd887c43e5341f3e68ffe47f0f5a">mlx::core::topk</a></div><div class="ttdeci">array topk(const array &a, int k, StreamOrDevice s={})</div><div class="ttdoc">Returns topk elements of the flattened array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga54ca54f06bfb2be15b163a5209e2a0f0"><div class="ttname"><a href="group__ops.html#ga54ca54f06bfb2be15b163a5209e2a0f0">mlx::core::expm1</a></div><div class="ttdeci">array expm1(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Computes the expm1 function of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga54eeed455321a54c8e72e16552a978f2"><div class="ttname"><a href="group__ops.html#ga54eeed455321a54c8e72e16552a978f2">mlx::core::ones</a></div><div class="ttdeci">array ones(const std::vector< int > &shape, Dtype dtype, StreamOrDevice s={})</div><div class="ttdoc">Fill an array of the given shape with ones.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga5528e80f5e8bad71e106a0cf9edd8920"><div class="ttname"><a href="group__ops.html#ga5528e80f5e8bad71e106a0cf9edd8920">mlx::core::abs</a></div><div class="ttdeci">array abs(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Absolute value of elements in an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga577c911618575314de63d1060656a26e"><div class="ttname"><a href="group__ops.html#ga577c911618575314de63d1060656a26e">mlx::core::meshgrid</a></div><div class="ttdeci">std::vector< array > meshgrid(const std::vector< array > &arrays, bool sparse=false, std::string indexing="xy", StreamOrDevice s={})</div><div class="ttdoc">A vector of coordinate arrays from coordinate vectors.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga5b596906bf8cdc8d97ed6ddc9aeb4c23"><div class="ttname"><a href="group__ops.html#ga5b596906bf8cdc8d97ed6ddc9aeb4c23">mlx::core::conjugate</a></div><div class="ttdeci">array conjugate(const array &a, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga5efb19aa0dfa42d8a3d5e1dfd569cd6d"><div class="ttname"><a href="group__ops.html#ga5efb19aa0dfa42d8a3d5e1dfd569cd6d">mlx::core::tanh</a></div><div class="ttdeci">array tanh(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Hyperbolic Tangent of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga6381c7062b232ac21f3fdc3136aaf86e"><div class="ttname"><a href="group__ops.html#ga6381c7062b232ac21f3fdc3136aaf86e">mlx::core::quantized_matmul</a></div><div class="ttdeci">array quantized_matmul(const array &x, const array &w, const array &scales, const array &biases, bool transpose=true, int group_size=64, int bits=4, StreamOrDevice s={})</div><div class="ttdoc">Quantized matmul multiplies x with a quantized matrix w.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga654fec16a9746b390916697a2ab2546e"><div class="ttname"><a href="group__ops.html#ga654fec16a9746b390916697a2ab2546e">mlx::core::inner</a></div><div class="ttdeci">array inner(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Compute the inner product of two vectors.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga6b76c8ea46b19e6866af155fa5910be6"><div class="ttname"><a href="group__ops.html#ga6b76c8ea46b19e6866af155fa5910be6">mlx::core::block_masked_mm</a></div><div class="ttdeci">array block_masked_mm(array a, array b, int block_size, std::optional< array > mask_out=std::nullopt, std::optional< array > mask_lhs=std::nullopt, std::optional< array > mask_rhs=std::nullopt, StreamOrDevice s={})</div><div class="ttdoc">Compute matrix product with block masking.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga6caba9c92b5989123501f909cc7da354"><div class="ttname"><a href="group__ops.html#ga6caba9c92b5989123501f909cc7da354">mlx::core::arctan2</a></div><div class="ttdeci">array arctan2(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Inverse tangent of the ratio of two arrays.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga6d5f5f72362488b956cdc4615ef6c636"><div class="ttname"><a href="group__ops.html#ga6d5f5f72362488b956cdc4615ef6c636">mlx::core::number_of_elements</a></div><div class="ttdeci">array number_of_elements(const array &a, std::vector< int > axes, bool inverted, Dtype dtype=int32, StreamOrDevice s={})</div><div class="ttdoc">Extract the number of elements along some axes as a scalar array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga6e9907d2f14dc4803e4306b3dbc4b3ca"><div class="ttname"><a href="group__ops.html#ga6e9907d2f14dc4803e4306b3dbc4b3ca">mlx::core::conv3d</a></div><div class="ttdeci">array conv3d(const array &input, const array &weight, const std::tuple< int, int, int > &stride={1, 1, 1}, const std::tuple< int, int, int > &padding={0, 0, 0}, const std::tuple< int, int, int > &dilation={1, 1, 1}, int groups=1, StreamOrDevice s={})</div><div class="ttdoc">3D convolution with a filter</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga6fb22d4926133573e430fcc92f4eef31"><div class="ttname"><a href="group__ops.html#ga6fb22d4926133573e430fcc92f4eef31">mlx::core::log</a></div><div class="ttdeci">array log(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Natural logarithm of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga708abf8f79609cd6831db7c38cafac0e"><div class="ttname"><a href="group__ops.html#ga708abf8f79609cd6831db7c38cafac0e">mlx::core::sigmoid</a></div><div class="ttdeci">array sigmoid(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Element-wise logistic sigmoid of the array: 1 / (1 + exp(-x).</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga710daa7ec721bd4d3f326082cb195576"><div class="ttname"><a href="group__ops.html#ga710daa7ec721bd4d3f326082cb195576">mlx::core::squeeze</a></div><div class="ttdeci">array squeeze(const array &a, const std::vector< int > &axes, StreamOrDevice s={})</div><div class="ttdoc">Remove singleton dimensions at the given axes.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga7153071bcfff6faad21332163fb9a430"><div class="ttname"><a href="group__ops.html#ga7153071bcfff6faad21332163fb9a430">mlx::core::greater_equal</a></div><div class="ttdeci">array greater_equal(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Returns bool array with (a >= b) element-wise.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga717f11149a8c7b4cc3e33bbcc0a97133"><div class="ttname"><a href="group__ops.html#ga717f11149a8c7b4cc3e33bbcc0a97133">mlx::core::expand_dims</a></div><div class="ttdeci">array expand_dims(const array &a, const std::vector< int > &axes, StreamOrDevice s={})</div><div class="ttdoc">Add a singleton dimension at the given axes.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga725ff0789f934b1fdd54ee29e47022ff"><div class="ttname"><a href="group__ops.html#ga725ff0789f934b1fdd54ee29e47022ff">mlx::core::isfinite</a></div><div class="ttdeci">array isfinite(const array &a, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga73b02833229678786e7f302d458d5a83"><div class="ttname"><a href="group__ops.html#ga73b02833229678786e7f302d458d5a83">mlx::core::conv2d</a></div><div class="ttdeci">array conv2d(const array &input, const array &weight, const std::pair< int, int > &stride={1, 1}, const std::pair< int, int > &padding={0, 0}, const std::pair< int, int > &dilation={1, 1}, int groups=1, StreamOrDevice s={})</div><div class="ttdoc">2D convolution with a filter</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga74fd2777adef10e6fe628a9cdadb01cb"><div class="ttname"><a href="group__ops.html#ga74fd2777adef10e6fe628a9cdadb01cb">mlx::core::operator></a></div><div class="ttdeci">array operator>(const array &a, const array &b)</div><div class="ttdef"><b>Definition</b> ops.h:331</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga752fd2707dabb05d0308ba3d55346ada"><div class="ttname"><a href="group__ops.html#ga752fd2707dabb05d0308ba3d55346ada">mlx::core::bitwise_and</a></div><div class="ttdeci">array bitwise_and(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Bitwise and.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga7534290bceab5fb3831a05d67bebce7d"><div class="ttname"><a href="group__ops.html#ga7534290bceab5fb3831a05d67bebce7d">mlx::core::split</a></div><div class="ttdeci">std::vector< array > split(const array &a, int num_splits, int axis, StreamOrDevice s={})</div><div class="ttdoc">Split an array into sub-arrays along a given axis.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga753d59f5a9f5f2362865ee83b4dced2a"><div class="ttname"><a href="group__ops.html#ga753d59f5a9f5f2362865ee83b4dced2a">mlx::core::matmul</a></div><div class="ttdeci">array matmul(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Matrix-matrix multiplication.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga768977cda8d68cf23f464a6af9907876"><div class="ttname"><a href="group__ops.html#ga768977cda8d68cf23f464a6af9907876">mlx::core::logical_and</a></div><div class="ttdeci">array logical_and(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Logical and of two arrays.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga76fb9062c64264e34d2e07013390557c"><div class="ttname"><a href="group__ops.html#ga76fb9062c64264e34d2e07013390557c">mlx::core::erfinv</a></div><div class="ttdeci">array erfinv(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Computes the inverse error function of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga77472dd06cfa7a30a42e4fd927bd859f"><div class="ttname"><a href="group__ops.html#ga77472dd06cfa7a30a42e4fd927bd859f">mlx::core::divide</a></div><div class="ttdeci">array divide(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Divide two arrays.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga7972058715c26559dff9c9ae2a3ef76d"><div class="ttname"><a href="group__ops.html#ga7972058715c26559dff9c9ae2a3ef76d">mlx::core::power</a></div><div class="ttdeci">array power(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Raise elements of a to the power of b element-wise.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga7ade2ea305e2e4219c3609443fb5db8d"><div class="ttname"><a href="group__ops.html#ga7ade2ea305e2e4219c3609443fb5db8d">mlx::core::maximum</a></div><div class="ttdeci">array maximum(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Element-wise maximum between two arrays.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga7b0e3d46a39d99b1628b422049790082"><div class="ttname"><a href="group__ops.html#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_ga7c3bd5ef430a71dfd298e626741e3c71"><div class="ttname"><a href="group__ops.html#ga7c3bd5ef430a71dfd298e626741e3c71">mlx::core::argmin</a></div><div class="ttdeci">array argmin(const array &a, bool keepdims, StreamOrDevice s={})</div><div class="ttdoc">Returns the index of the minimum value in the array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga7e133df686439588a8cd1fb10ce0c6e9"><div class="ttname"><a href="group__ops.html#ga7e133df686439588a8cd1fb10ce0c6e9">mlx::core::var</a></div><div class="ttdeci">array var(const array &a, bool keepdims, int ddof=0, StreamOrDevice s={})</div><div class="ttdoc">Computes the variance of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga7e85a476530c96ab73b894d1bddacf84"><div class="ttname"><a href="group__ops.html#ga7e85a476530c96ab73b894d1bddacf84">mlx::core::full</a></div><div class="ttdeci">array full(std::vector< int > shape, array vals, Dtype dtype, StreamOrDevice s={})</div><div class="ttdoc">Fill an array of the given shape with the given value(s).</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga7e9bb08b43c8fd0444b7d3c9e09dc1c6"><div class="ttname"><a href="group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6">mlx::core::softmax</a></div><div class="ttdeci">array softmax(const array &a, const std::vector< int > &axes, bool precise=false, StreamOrDevice s={})</div><div class="ttdoc">Softmax of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga7fb616054665b3c2d61fa234f501f079"><div class="ttname"><a href="group__ops.html#ga7fb616054665b3c2d61fa234f501f079">mlx::core::sort</a></div><div class="ttdeci">array sort(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Returns a sorted copy of the flattened array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga7fed87d96cc7741d8267f4eac83f5fe7"><div class="ttname"><a href="group__ops.html#ga7fed87d96cc7741d8267f4eac83f5fe7">mlx::core::max</a></div><div class="ttdeci">array max(const array &a, bool keepdims, StreamOrDevice s={})</div><div class="ttdoc">The maximum of all elements of the array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga7ff592a64d528f0cf4f3d098465da029"><div class="ttname"><a href="group__ops.html#ga7ff592a64d528f0cf4f3d098465da029">mlx::core::imag</a></div><div class="ttdeci">array imag(const array &a, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga81c2f138ea81d78affd42e9a646cacbc"><div class="ttname"><a href="group__ops.html#ga81c2f138ea81d78affd42e9a646cacbc">mlx::core::pad</a></div><div class="ttdeci">array pad(const array &a, const std::vector< int > &axes, const std::vector< int > &low_pad_size, const std::vector< int > &high_pad_size, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={})</div><div class="ttdoc">Pad an array with a constant value.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga82a53e083205a965387b3c3e2463244a"><div class="ttname"><a href="group__ops.html#ga82a53e083205a965387b3c3e2463244a">mlx::core::addmm</a></div><div class="ttdeci">array addmm(array c, array a, array b, const float &alpha=1.f, const float &beta=1.f, StreamOrDevice s={})</div><div class="ttdoc">Compute D = beta * C + alpha * (A @ B)</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga83e0bb45dc770cf014531d873b78c5a2"><div class="ttname"><a href="group__ops.html#ga83e0bb45dc770cf014531d873b78c5a2">mlx::core::tril</a></div><div class="ttdeci">array tril(array x, int k=0, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8598dd718fb05cb28535e250372d4e6f"><div class="ttname"><a href="group__ops.html#ga8598dd718fb05cb28535e250372d4e6f">mlx::core::any</a></div><div class="ttdeci">array any(const array &a, bool keepdims, StreamOrDevice s={})</div><div class="ttdoc">True if any elements in the array are true (or non-zero).</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga866af24e10db2797e1c5a5986dbf6c0d"><div class="ttname"><a href="group__ops.html#ga866af24e10db2797e1c5a5986dbf6c0d">mlx::core::outer</a></div><div class="ttdeci">array outer(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Compute the outer product of two vectors.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga872d2c1806e67ce2596b24d056681074"><div class="ttname"><a href="group__ops.html#ga872d2c1806e67ce2596b24d056681074">mlx::core::hadamard_transform</a></div><div class="ttdeci">array hadamard_transform(const array &a, std::optional< float > scale=std::nullopt, StreamOrDevice s={})</div><div class="ttdoc">Multiply the array by the Hadamard matrix of corresponding size.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8770e8c8f23f13343911f4c9d6e1c619"><div class="ttname"><a href="group__ops.html#ga8770e8c8f23f13343911f4c9d6e1c619">mlx::core::arcsin</a></div><div class="ttdeci">array arcsin(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Arc Sine of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga89682bf78491761e062d4ee7bef0c829"><div class="ttname"><a href="group__ops.html#ga89682bf78491761e062d4ee7bef0c829">mlx::core::left_shift</a></div><div class="ttdeci">array left_shift(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Shift bits to the left.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8a2056f8c9bb30914c40bcf509386491"><div class="ttname"><a href="group__ops.html#ga8a2056f8c9bb30914c40bcf509386491">mlx::core::where</a></div><div class="ttdeci">array where(const array &condition, const array &x, const array &y, StreamOrDevice s={})</div><div class="ttdoc">Select from x or y depending on condition.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8a3b04e23e347d99ecf411fd6f4e5125"><div class="ttname"><a href="group__ops.html#ga8a3b04e23e347d99ecf411fd6f4e5125">mlx::core::exp</a></div><div class="ttdeci">array exp(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Exponential of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8af4f22c08c11c4ffab7e3d45e0f3cd6"><div class="ttname"><a href="group__ops.html#ga8af4f22c08c11c4ffab7e3d45e0f3cd6">mlx::core::bitwise_or</a></div><div class="ttdeci">array bitwise_or(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Bitwise inclusive or.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8d50480266d258cac40ff51bcb0fc6a7"><div class="ttname"><a href="group__ops.html#ga8d50480266d258cac40ff51bcb0fc6a7">mlx::core::gather_mm</a></div><div class="ttdeci">array gather_mm(array a, array b, std::optional< array > lhs_indices=std::nullopt, std::optional< array > rhs_indices=std::nullopt, StreamOrDevice s={})</div><div class="ttdoc">Compute matrix product with matrix-level gather.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8d656904aa2690b60955ae745aecfc30"><div class="ttname"><a href="group__ops.html#ga8d656904aa2690b60955ae745aecfc30">mlx::core::floor</a></div><div class="ttdeci">array floor(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Floor the element of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8db814da631d9cd32a8d6563bf4ac530"><div class="ttname"><a href="group__ops.html#ga8db814da631d9cd32a8d6563bf4ac530">mlx::core::conv_transpose3d</a></div><div class="ttdeci">array conv_transpose3d(const array &input, const array &weight, const std::tuple< int, int, int > &stride={1, 1, 1}, const std::tuple< int, int, int > &padding={0, 0, 0}, const std::tuple< int, int, int > &dilation={1, 1, 1}, int groups=1, StreamOrDevice s={})</div><div class="ttdoc">3D transposed convolution with a filter</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8de80ecef30fc560003d40f61a38b99d"><div class="ttname"><a href="group__ops.html#ga8de80ecef30fc560003d40f61a38b99d">mlx::core::as_strided</a></div><div class="ttdeci">array as_strided(array a, std::vector< int > shape, std::vector< size_t > strides, size_t offset, StreamOrDevice s={})</div><div class="ttdoc">Create a view of an array with the given shape and strides.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8df3b2703bf671457422894dd870cdc5"><div class="ttname"><a href="group__ops.html#ga8df3b2703bf671457422894dd870cdc5">mlx::core::argsort</a></div><div class="ttdeci">array argsort(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Returns indices that sort the flattened array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8e0caebf43cd65bd40e4ce97922cd06b"><div class="ttname"><a href="group__ops.html#ga8e0caebf43cd65bd40e4ce97922cd06b">mlx::core::put_along_axis</a></div><div class="ttdeci">array put_along_axis(const array &a, const array &indices, const array &values, int axis, StreamOrDevice s={})</div><div class="ttdoc">Put the values into the array at the given indices along the axis.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8f3059336ee0c87207b1f8c6ab312645"><div class="ttname"><a href="group__ops.html#ga8f3059336ee0c87207b1f8c6ab312645">mlx::core::array_equal</a></div><div class="ttdeci">array array_equal(const array &a, const array &b, bool equal_nan, StreamOrDevice s={})</div><div class="ttdoc">True if two arrays have the same shape and elements.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8fc238d5e5d1153e69da8b36015d9844"><div class="ttname"><a href="group__ops.html#ga8fc238d5e5d1153e69da8b36015d9844">mlx::core::isinf</a></div><div class="ttdeci">array isinf(const array &a, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga9142b8d717699a8abfa2a7398891ff8a"><div class="ttname"><a href="group__ops.html#ga9142b8d717699a8abfa2a7398891ff8a">mlx::core::less</a></div><div class="ttdeci">array less(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Returns bool array with (a < b) element-wise.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga9236b085a88ead3128ed8079d009cac6"><div class="ttname"><a href="group__ops.html#ga9236b085a88ead3128ed8079d009cac6">mlx::core::diagonal</a></div><div class="ttdeci">array diagonal(const array &a, int offset=0, int axis1=0, int axis2=1, StreamOrDevice s={})</div><div class="ttdoc">Extract a diagonal or construct a diagonal array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga94f8d3b1906fee99da9cbe39f7be7d42"><div class="ttname"><a href="group__ops.html#ga94f8d3b1906fee99da9cbe39f7be7d42">mlx::core::ones_like</a></div><div class="ttdeci">array ones_like(const array &a, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga95d9a9425533b5ed1707eb00184dffc6"><div class="ttname"><a href="group__ops.html#ga95d9a9425533b5ed1707eb00184dffc6">mlx::core::negative</a></div><div class="ttdeci">array negative(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Negate an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga968bcabed902311dcfbd903b0fb886ec"><div class="ttname"><a href="group__ops.html#ga968bcabed902311dcfbd903b0fb886ec">mlx::core::linspace</a></div><div class="ttdeci">array linspace(double start, double stop, int num=50, Dtype dtype=float32, StreamOrDevice s={})</div><div class="ttdoc">A 1D array of num evenly spaced numbers in the range [start, stop]</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga99f5c904f724156a814d7817188351d2"><div class="ttname"><a href="group__ops.html#ga99f5c904f724156a814d7817188351d2">mlx::core::remainder</a></div><div class="ttdeci">array remainder(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Compute the element-wise remainder of division.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaa041f3f070e68f4946db07516b7d092e"><div class="ttname"><a href="group__ops.html#gaa041f3f070e68f4946db07516b7d092e">mlx::core::arctan</a></div><div class="ttdeci">array arctan(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Arc Tangent of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaa30bf1adcd78d1c2595d07b215731714"><div class="ttname"><a href="group__ops.html#gaa30bf1adcd78d1c2595d07b215731714">mlx::core::conv_transpose1d</a></div><div class="ttdeci">array conv_transpose1d(const array &input, const array &weight, int stride=1, int padding=0, int dilation=1, int groups=1, StreamOrDevice s={})</div><div class="ttdoc">1D transposed convolution with a filter</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaa30ebc0a8376dbc3f7e46a47052b5894"><div class="ttname"><a href="group__ops.html#gaa30ebc0a8376dbc3f7e46a47052b5894">mlx::core::divmod</a></div><div class="ttdeci">std::vector< array > divmod(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Compute the element-wise quotient and remainder.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaa9df5917876eeb0cb28b7fa81f880412"><div class="ttname"><a href="group__ops.html#gaa9df5917876eeb0cb28b7fa81f880412">mlx::core::triu</a></div><div class="ttdeci">array triu(array x, int k=0, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaafafcfcebdf7248679c8543d0c0497e5"><div class="ttname"><a href="group__ops.html#gaafafcfcebdf7248679c8543d0c0497e5">mlx::core::arccosh</a></div><div class="ttdeci">array arccosh(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Inverse Hyperbolic Cosine of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gab105a57b9a4d84496fe1e4d60e13d361"><div class="ttname"><a href="group__ops.html#gab105a57b9a4d84496fe1e4d60e13d361">mlx::core::tile</a></div><div class="ttdeci">array tile(const array &arr, std::vector< int > reps, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gab1467c6a9e675152e768afd6dcfb61de"><div class="ttname"><a href="group__ops.html#gab1467c6a9e675152e768afd6dcfb61de">mlx::core::nan_to_num</a></div><div class="ttdeci">array nan_to_num(const array &a, float nan=0.0f, const std::optional< float > posinf=std::nullopt, const std::optional< float > neginf=std::nullopt, StreamOrDevice s={})</div><div class="ttdoc">Replace NaN and infinities with finite numbers.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gab27599802617a4c8f9964ab5f4ffee12"><div class="ttname"><a href="group__ops.html#gab27599802617a4c8f9964ab5f4ffee12">mlx::core::min</a></div><div class="ttdeci">array min(const array &a, bool keepdims, StreamOrDevice s={})</div><div class="ttdoc">The minimum of all elements of the array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gab3bfbf82b1e4de7b00bbcf1a2255fbde"><div class="ttname"><a href="group__ops.html#gab3bfbf82b1e4de7b00bbcf1a2255fbde">mlx::core::operator%</a></div><div class="ttdeci">array operator%(const array &a, const array &b)</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gab43cc28690da7cdd43b43065adbd31da"><div class="ttname"><a href="group__ops.html#gab43cc28690da7cdd43b43065adbd31da">mlx::core::quantize</a></div><div class="ttdeci">std::tuple< array, array, array > quantize(const array &w, int group_size=64, int bits=4, StreamOrDevice s={})</div><div class="ttdoc">Quantize a matrix along its last axis.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gab46a35925a04c5a9d2ec7898ee55358e"><div class="ttname"><a href="group__ops.html#gab46a35925a04c5a9d2ec7898ee55358e">mlx::core::arctanh</a></div><div class="ttdeci">array arctanh(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Inverse Hyperbolic Tangent of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gab49e3a687e826554ed1574186e8ae974"><div class="ttname"><a href="group__ops.html#gab49e3a687e826554ed1574186e8ae974">mlx::core::repeat</a></div><div class="ttdeci">array repeat(const array &arr, int repeats, int axis, StreamOrDevice s={})</div><div class="ttdoc">Repeat an array along an axis.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gab6e7f655a9ff15350ca5379692f9d444"><div class="ttname"><a href="group__ops.html#gab6e7f655a9ff15350ca5379692f9d444">mlx::core::gather</a></div><div class="ttdeci">array gather(const array &a, const std::vector< array > &indices, const std::vector< int > &axes, const std::vector< int > &slice_sizes, StreamOrDevice s={})</div><div class="ttdoc">Gather array entries given indices and slices.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gab783890428b596f715dc7dd2057eae99"><div class="ttname"><a href="group__ops.html#gab783890428b596f715dc7dd2057eae99">mlx::core::broadcast_arrays</a></div><div class="ttdeci">std::vector< array > broadcast_arrays(const std::vector< array > &inputs, StreamOrDevice s={})</div><div class="ttdoc">Broadcast a vector of arrays against one another.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaba4d25e7a2bf87ba4feb7837ec7fa396"><div class="ttname"><a href="group__ops.html#gaba4d25e7a2bf87ba4feb7837ec7fa396">mlx::core::atleast_1d</a></div><div class="ttdeci">array atleast_1d(const array &a, StreamOrDevice s={})</div><div class="ttdoc">convert an array to an atleast ndim array</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gabc46eed81ab6c6247903e4ec0c4ec1fb"><div class="ttname"><a href="group__ops.html#gabc46eed81ab6c6247903e4ec0c4ec1fb">mlx::core::swapaxes</a></div><div class="ttdeci">array swapaxes(const array &a, int axis1, int axis2, StreamOrDevice s={})</div><div class="ttdoc">Swap two axes of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gabca78d34ce93f0de2814e62225bb2a53"><div class="ttname"><a href="group__ops.html#gabca78d34ce93f0de2814e62225bb2a53">mlx::core::logical_not</a></div><div class="ttdeci">array logical_not(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Logical not of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gabdc36fa65697d0361c8d67495de77129"><div class="ttname"><a href="group__ops.html#gabdc36fa65697d0361c8d67495de77129">mlx::core::concatenate</a></div><div class="ttdeci">array concatenate(const std::vector< array > &arrays, int axis, StreamOrDevice s={})</div><div class="ttdoc">Concatenate arrays along a given axis.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gabf786129c7660ed8d5acb5499bc6fefd"><div class="ttname"><a href="group__ops.html#gabf786129c7660ed8d5acb5499bc6fefd">mlx::core::trace</a></div><div class="ttdeci">array trace(const array &a, int offset, int axis1, int axis2, Dtype dtype, StreamOrDevice s={})</div><div class="ttdoc">Return the sum along a specified diagonal in the given array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gabff758a5c1ce32ad7e8b78aba0164077"><div class="ttname"><a href="group__ops.html#gabff758a5c1ce32ad7e8b78aba0164077">mlx::core::dequantize</a></div><div class="ttdeci">array dequantize(const array &w, const array &scales, const array &biases, int group_size=64, int bits=4, StreamOrDevice s={})</div><div class="ttdoc">Dequantize a matrix produced by quantize()</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gac1869f3b7094869b44fe7ac4ce58638b"><div class="ttname"><a href="group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b">mlx::core::transpose</a></div><div class="ttdeci">array transpose(const array &a, std::vector< int > axes, StreamOrDevice s={})</div><div class="ttdoc">Permutes the dimensions according to the given axes.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gac1b30830a972fb9a2601379ad2b32405"><div class="ttname"><a href="group__ops.html#gac1b30830a972fb9a2601379ad2b32405">mlx::core::partition</a></div><div class="ttdeci">array partition(const array &a, int kth, StreamOrDevice s={})</div><div class="ttdoc">Returns a partitioned copy of the flattened array such that the smaller kth elements are first.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gac2fc270882fcfa81eb8bd068cc0d86d7"><div class="ttname"><a href="group__ops.html#gac2fc270882fcfa81eb8bd068cc0d86d7">mlx::core::take</a></div><div class="ttdeci">array take(const array &a, const array &indices, int axis, StreamOrDevice s={})</div><div class="ttdoc">Take array slices at the given indices of the specified axis.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gac3a6fe18694e84b3d63458e9553ac181"><div class="ttname"><a href="group__ops.html#gac3a6fe18694e84b3d63458e9553ac181">mlx::core::operator^</a></div><div class="ttdeci">array operator^(const array &a, const array &b)</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gac40e48c69f9c715a767912c30836e75c"><div class="ttname"><a href="group__ops.html#gac40e48c69f9c715a767912c30836e75c">mlx::core::roll</a></div><div class="ttdeci">array roll(const array &a, int shift, StreamOrDevice s={})</div><div class="ttdoc">Roll elements along an axis and introduce them on the other side.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gac4a51a68fbe1725436b026d2fbb95759"><div class="ttname"><a href="group__ops.html#gac4a51a68fbe1725436b026d2fbb95759">mlx::core::depends</a></div><div class="ttdeci">std::vector< array > depends(const std::vector< array > &inputs, const std::vector< array > &dependencies)</div><div class="ttdoc">Implements the identity function but allows injecting dependencies to other arrays.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gac62e2cedc49ef2c90dd8584000317450"><div class="ttname"><a href="group__ops.html#gac62e2cedc49ef2c90dd8584000317450">mlx::core::arcsinh</a></div><div class="ttdeci">array arcsinh(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Inverse Hyperbolic Sine of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gacd14c2b5cfebf343fc2d672722f8d174"><div class="ttname"><a href="group__ops.html#gacd14c2b5cfebf343fc2d672722f8d174">mlx::core::scatter_add</a></div><div class="ttdeci">array scatter_add(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})</div><div class="ttdoc">Scatter and add updates to given indices.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gacff4eb57c085d571e722083680267ac5"><div class="ttname"><a href="group__ops.html#gacff4eb57c085d571e722083680267ac5">mlx::core::logsumexp</a></div><div class="ttdeci">array logsumexp(const array &a, bool keepdims, StreamOrDevice s={})</div><div class="ttdoc">The logsumexp of all elements of the array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gad256e86cc1a6e6b3832e392baa90318d"><div class="ttname"><a href="group__ops.html#gad256e86cc1a6e6b3832e392baa90318d">mlx::core::broadcast_to</a></div><div class="ttdeci">array broadcast_to(const array &a, const std::vector< int > &shape, StreamOrDevice s={})</div><div class="ttdoc">Broadcast an array to a given shape.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gad438be8f90bae9d37c6853b8f4225d61"><div class="ttname"><a href="group__ops.html#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_gad656c30f9fd7d9467e405657b325aa7e"><div class="ttname"><a href="group__ops.html#gad656c30f9fd7d9467e405657b325aa7e">mlx::core::operator<<</a></div><div class="ttdeci">array operator<<(const array &a, const array &b)</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gad66135407dbb41b3c5d2cdfd51226c21"><div class="ttname"><a href="group__ops.html#gad66135407dbb41b3c5d2cdfd51226c21">mlx::core::slice</a></div><div class="ttdeci">array slice(const array &a, std::vector< int > start, std::vector< int > stop, std::vector< int > strides, StreamOrDevice s={})</div><div class="ttdoc">Slice an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gad80f7c4a58c12b6cb30a8b9a73008993"><div class="ttname"><a href="group__ops.html#gad80f7c4a58c12b6cb30a8b9a73008993">mlx::core::isposinf</a></div><div class="ttdeci">array isposinf(const array &a, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaddc825a5c173e195ab0fda83ad630420"><div class="ttname"><a href="group__ops.html#gaddc825a5c173e195ab0fda83ad630420">mlx::core::cumsum</a></div><div class="ttdeci">array cumsum(const array &a, int axis, bool reverse=false, bool inclusive=true, StreamOrDevice s={})</div><div class="ttdoc">Cumulative sum of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gade2eea48989f4caaf36e89f7bd2a8816"><div class="ttname"><a href="group__ops.html#gade2eea48989f4caaf36e89f7bd2a8816">mlx::core::operator-</a></div><div class="ttdeci">array operator-(const array &a)</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gade46e768fd46b8b640eb16f26abeecef"><div class="ttname"><a href="group__ops.html#gade46e768fd46b8b640eb16f26abeecef">mlx::core::mean</a></div><div class="ttdeci">array mean(const array &a, bool keepdims, StreamOrDevice s={})</div><div class="ttdoc">Computes the mean of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gade905ee92eb6ab7edfc312aeddfbaeb6"><div class="ttname"><a href="group__ops.html#gade905ee92eb6ab7edfc312aeddfbaeb6">mlx::core::sum</a></div><div class="ttdeci">array sum(const array &a, bool keepdims, StreamOrDevice s={})</div><div class="ttdoc">Sums the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gae0a81d4f983e296a87302e36d65bfc76"><div class="ttname"><a href="group__ops.html#gae0a81d4f983e296a87302e36d65bfc76">mlx::core::take_along_axis</a></div><div class="ttdeci">array take_along_axis(const array &a, const array &indices, int axis, StreamOrDevice s={})</div><div class="ttdoc">Take array entries given indices along the axis.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gae60b0b5339b9c50b9970260faf613e83"><div class="ttname"><a href="group__ops.html#gae60b0b5339b9c50b9970260faf613e83">mlx::core::argmax</a></div><div class="ttdeci">array argmax(const array &a, bool keepdims, StreamOrDevice s={})</div><div class="ttdoc">Returns the index of the maximum value in the array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaebb59971cb9bc45005dc1d398e4f0a3d"><div class="ttname"><a href="group__ops.html#gaebb59971cb9bc45005dc1d398e4f0a3d">mlx::core::conv_transpose2d</a></div><div class="ttdeci">array conv_transpose2d(const array &input, const array &weight, const std::pair< int, int > &stride={1, 1}, const std::pair< int, int > &padding={0, 0}, const std::pair< int, int > &dilation={1, 1}, int groups=1, StreamOrDevice s={})</div><div class="ttdoc">2D transposed convolution with a filter</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaebf0a73ad3732fba39df37826c235692"><div class="ttname"><a href="group__ops.html#gaebf0a73ad3732fba39df37826c235692">mlx::core::sin</a></div><div class="ttdeci">array sin(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Sine of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaee1d774bb0843601d7a0a4257d616ae3"><div class="ttname"><a href="group__ops.html#gaee1d774bb0843601d7a0a4257d616ae3">mlx::core::operator&&</a></div><div class="ttdeci">array operator&&(const array &a, const array &b)</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaee37cac8476e8f8d666bcded5bc59143"><div class="ttname"><a href="group__ops.html#gaee37cac8476e8f8d666bcded5bc59143">mlx::core::cummax</a></div><div class="ttdeci">array cummax(const array &a, int axis, bool reverse=false, bool inclusive=true, StreamOrDevice s={})</div><div class="ttdoc">Cumulative max of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaee41e2b8f61d563200ff03575ac1d6c3"><div class="ttname"><a href="group__ops.html#gaee41e2b8f61d563200ff03575ac1d6c3">mlx::core::operator<</a></div><div class="ttdeci">array operator<(const array &a, const array &b)</div><div class="ttdef"><b>Definition</b> ops.h:359</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaeeb7f5bb88aa32a3ac2be1f39c5f8087"><div class="ttname"><a href="group__ops.html#gaeeb7f5bb88aa32a3ac2be1f39c5f8087">mlx::core::atleast_2d</a></div><div class="ttdeci">array atleast_2d(const array &a, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaeedf77f722b394429f1a7f6c367883bf"><div class="ttname"><a href="group__ops.html#gaeedf77f722b394429f1a7f6c367883bf">mlx::core::operator/</a></div><div class="ttdeci">array operator/(const array &a, const array &b)</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaf0cd4257de7542daf9faf5e605e31020"><div class="ttname"><a href="group__ops.html#gaf0cd4257de7542daf9faf5e605e31020">mlx::core::allclose</a></div><div class="ttdeci">array allclose(const array &a, const array &b, double rtol=1e-5, double atol=1e-8, bool equal_nan=false, StreamOrDevice s={})</div><div class="ttdoc">True if the two arrays are equal within the specified tolerance.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaf0d232de4cbfffda1e2c838f8afdf6ff"><div class="ttname"><a href="group__ops.html#gaf0d232de4cbfffda1e2c838f8afdf6ff">mlx::core::operator&</a></div><div class="ttdeci">array operator&(const array &a, const array &b)</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaf301c49c10fa9b95a9e8dc52ead1a8dd"><div class="ttname"><a href="group__ops.html#gaf301c49c10fa9b95a9e8dc52ead1a8dd">mlx::core::argpartition</a></div><div class="ttdeci">array argpartition(const array &a, int kth, StreamOrDevice s={})</div><div class="ttdoc">Returns indices that partition the flattened array such that the smaller kth elements are first.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaf4ec7bfc1ad13b891f1f3ef1772ef04d"><div class="ttname"><a href="group__ops.html#gaf4ec7bfc1ad13b891f1f3ef1772ef04d">mlx::core::greater</a></div><div class="ttdeci">array greater(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Returns bool array with (a > b) element-wise.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaf532375c6563dbd6e329bdedf0224dd7"><div class="ttname"><a href="group__ops.html#gaf532375c6563dbd6e329bdedf0224dd7">mlx::core::sinh</a></div><div class="ttdeci">array sinh(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Hyperbolic Sine of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaf57392e641640b5d06e4c99518391c38"><div class="ttname"><a href="group__ops.html#gaf57392e641640b5d06e4c99518391c38">mlx::core::multiply</a></div><div class="ttdeci">array multiply(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Multiply two arrays.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaf5c9735f4690327e1500e04e728fae70"><div class="ttname"><a href="group__ops.html#gaf5c9735f4690327e1500e04e728fae70">mlx::core::tensordot</a></div><div class="ttdeci">array tensordot(const array &a, const array &b, const int axis=2, StreamOrDevice s={})</div><div class="ttdoc">Returns a contraction of a and b over multiple dimensions.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaf8913cabeb9fb193ba687aaeb2087764"><div class="ttname"><a href="group__ops.html#gaf8913cabeb9fb193ba687aaeb2087764">mlx::core::real</a></div><div class="ttdeci">array real(const array &a, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaf8f2ec2b98a4b59eca73d7471df6e032"><div class="ttname"><a href="group__ops.html#gaf8f2ec2b98a4b59eca73d7471df6e032">mlx::core::stack</a></div><div class="ttdeci">array stack(const std::vector< array > &arrays, int axis, StreamOrDevice s={})</div><div class="ttdoc">Stack arrays along a new axis.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaf985df6609c6bd75a14a844655d89eaa"><div class="ttname"><a href="group__ops.html#gaf985df6609c6bd75a14a844655d89eaa">mlx::core::logaddexp</a></div><div class="ttdeci">array logaddexp(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Log-add-exp of one elements in the array: log(exp(a) + exp(b)).</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gafa376ad57d38ba87378f0272dc379b23"><div class="ttname"><a href="group__ops.html#gafa376ad57d38ba87378f0272dc379b23">mlx::core::right_shift</a></div><div class="ttdeci">array right_shift(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Shift bits to the right.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gafbb857094d784b38c78683a091ffdbde"><div class="ttname"><a href="group__ops.html#gafbb857094d784b38c78683a091ffdbde">mlx::core::zeros_like</a></div><div class="ttdeci">array zeros_like(const array &a, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="anamespacemlx_1_1core_html"><div class="ttname"><a href="namespacemlx_1_1core.html">mlx::core</a></div><div class="ttdef"><b>Definition</b> allocator.h:7</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_a4734a596e57434492ddfe79f2cb9dbf9"><div class="ttname"><a href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">mlx::core::to_stream</a></div><div class="ttdeci">Stream to_stream(StreamOrDevice s)</div></div>
|
|
<div class="ttc" id="anamespacemlx_1_1core_html_a479648542a2bea151b947b18f0e79dd2"><div class="ttname"><a href="namespacemlx_1_1core.html#a479648542a2bea151b947b18f0e79dd2">mlx::core::copy</a></div><div class="ttdeci">void copy(const array &src, array &dst, CopyType ctype)</div></div>
|
|
<div class="ttc" id="anamespacemlx_1_1core_html_a5d6373aad1444edc9de1eb07bfe5cad3"><div class="ttname"><a href="namespacemlx_1_1core.html#a5d6373aad1444edc9de1eb07bfe5cad3">mlx::core::int32</a></div><div class="ttdeci">constexpr Dtype int32</div><div class="ttdef"><b>Definition</b> dtype.h:76</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 class="ttc" id="anamespacemlx_1_1core_html_a937503d72b66c661bf3f5fdcd98ef97c"><div class="ttname"><a href="namespacemlx_1_1core.html#a937503d72b66c661bf3f5fdcd98ef97c">mlx::core::operator==</a></div><div class="ttdeci">bool operator==(const Device &lhs, const Device &rhs)</div></div>
|
|
<div class="ttc" id="anamespacemlx_1_1core_html_a94d00a1b7f8a4717ab3f26f45e4da655"><div class="ttname"><a href="namespacemlx_1_1core.html#a94d00a1b7f8a4717ab3f26f45e4da655">mlx::core::operator!=</a></div><div class="ttdeci">bool operator!=(const Device &lhs, const Device &rhs)</div></div>
|
|
<div class="ttc" id="anamespacemlx_1_1core_html_a95fc1013cc48fbfee0c54310711a5e58"><div class="ttname"><a href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">mlx::core::StreamOrDevice</a></div><div class="ttdeci">std::variant< std::monostate, Stream, Device > StreamOrDevice</div><div class="ttdef"><b>Definition</b> utils.h:14</div></div>
|
|
<div class="ttc" id="astream_8h_html"><div class="ttname"><a href="stream_8h.html">stream.h</a></div></div>
|
|
<div class="ttc" id="astructmlx_1_1core_1_1_dtype_html"><div class="ttname"><a href="structmlx_1_1core_1_1_dtype.html">mlx::core::Dtype</a></div><div class="ttdef"><b>Definition</b> dtype.h:13</div></div>
|
|
<div class="ttc" id="autils_8h_html"><div class="ttname"><a href="utils_8h.html">utils.h</a></div></div>
|
|
</div><!-- fragment --></div><!-- contents -->
|
|
<!-- start footer part -->
|
|
<hr class="footer"/><address class="footer"><small>
|
|
Generated by <a href="https://www.doxygen.org/index.html"><img class="footer" src="doxygen.svg" width="104" height="31" alt="doxygen"/></a> 1.12.0
|
|
</small></address>
|
|
</div><!-- doc-content -->
|
|
</body>
|
|
</html>
|