2024-06-07 11:28:06 +08:00
<!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.10.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 >
< 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.10.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" 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');
$(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() { codefold.init(0); });
/* @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 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 >
2024-07-26 02:59:11 +08:00
< 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 >
2024-06-07 11:28:06 +08:00
< 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 >
2024-07-26 02:59:11 +08:00
< div class = "line" > < a id = "l00137" name = "l00137" > < / a > < span class = "lineno" > 137< / span > std::optional< float> scale = std::nullopt,< / div >
2024-06-07 11:28:06 +08:00
< 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 >
2024-07-12 06:32:08 +08:00
< 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 = "foldopen" id = "foldopen00147" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00147" name = "l00147" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga700dd51b77379a3d2260a55783e8ebf3" > 147< / 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#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 > < span class = "keywordflow" > return< / span > < a class = "code hl_function" href = "group__ops.html#ga710daa7ec721bd4d3f326082cb195576" > squeeze< / a > (a, std::vector< int> {axis}, s);< / div >
< div class = "line" > < a id = "l00149" name = "l00149" > < / a > < span class = "lineno" > 149< / span > }< / div >
< / div >
< div class = "line" > < a id = "l00150" name = "l00150" > < / a > < span class = "lineno" > 150< / span > < / div >
< div class = "line" > < a id = "l00152" name = "l00152" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga58bad3c61fd85b95927a987ba1cf5dad" > 152< / 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 >
2024-06-07 11:28:06 +08:00
< div class = "line" > < a id = "l00153" name = "l00153" > < / a > < span class = "lineno" > 153< / span > < / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00155" name = "l00155" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga717f11149a8c7b4cc3e33bbcc0a97133" > 155< / 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 = "l00156" name = "l00156" > < / a > < span class = "lineno" > 156< / 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 = "l00157" name = "l00157" > < / a > < span class = "lineno" > 157< / span > < span class = "keyword" > const< / span > std::vector< int> & axes,< / div >
< div class = "line" > < a id = "l00158" name = "l00158" > < / a > < span class = "lineno" > 158< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00159" name = "l00159" > < / a > < span class = "lineno" > 159< / span > < / div >
< div class = "line" > < a id = "l00161" name = "l00161" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga7a80adb4a5a36d18b5f234d4b034950a" > 161< / 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 = "l00162" name = "l00162" > < / a > < span class = "lineno" > 162< / span > < / div >
< div class = "line" > < a id = "l00164" name = "l00164" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gad66135407dbb41b3c5d2cdfd51226c21" > 164< / 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 = "l00165" name = "l00165" > < / a > < span class = "lineno" > 165< / 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 = "l00166" name = "l00166" > < / a > < span class = "lineno" > 166< / span > std::vector< int> start,< / div >
< div class = "line" > < a id = "l00167" name = "l00167" > < / a > < span class = "lineno" > 167< / span > std::vector< int> stop,< / div >
< div class = "line" > < a id = "l00168" name = "l00168" > < / a > < span class = "lineno" > 168< / span > std::vector< int> strides,< / div >
< div class = "line" > < a id = "l00169" name = "l00169" > < / a > < span class = "lineno" > 169< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00170" name = "l00170" > < / a > < span class = "lineno" > 170< / span > < / div >
< div class = "line" > < a id = "l00172" name = "l00172" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga90b243e25b5fbbd7a9732f7ac48f6403" > 172< / 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 = "l00173" name = "l00173" > < / a > < span class = "lineno" > 173< / 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 = "l00174" name = "l00174" > < / a > < span class = "lineno" > 174< / span > < span class = "keyword" > const< / span > std::vector< int> & start,< / div >
< div class = "line" > < a id = "l00175" name = "l00175" > < / a > < span class = "lineno" > 175< / span > < span class = "keyword" > const< / span > std::vector< int> & stop,< / div >
< div class = "line" > < a id = "l00176" name = "l00176" > < / a > < span class = "lineno" > 176< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00177" name = "l00177" > < / a > < span class = "lineno" > 177< / span > < / div >
< div class = "line" > < a id = "l00179" name = "l00179" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga3875660e4ce2c8add8bfcf8144078708" > 179< / 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 = "l00180" name = "l00180" > < / a > < span class = "lineno" > 180< / 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 = "l00181" name = "l00181" > < / a > < span class = "lineno" > 181< / 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 = "l00182" name = "l00182" > < / a > < span class = "lineno" > 182< / span > std::vector< int> start,< / div >
< div class = "line" > < a id = "l00183" name = "l00183" > < / a > < span class = "lineno" > 183< / span > std::vector< int> stop,< / div >
< div class = "line" > < a id = "l00184" name = "l00184" > < / a > < span class = "lineno" > 184< / span > std::vector< int> strides,< / div >
< div class = "line" > < a id = "l00185" name = "l00185" > < / a > < span class = "lineno" > 185< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00186" name = "l00186" > < / a > < span class = "lineno" > 186< / span > < / div >
< div class = "line" > < a id = "l00188" name = "l00188" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga03ffbbb4d989a463ef43f41ebf7eabef" > 188< / 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 = "l00189" name = "l00189" > < / a > < span class = "lineno" > 189< / 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 = "l00190" name = "l00190" > < / a > < span class = "lineno" > 190< / 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 = "l00191" name = "l00191" > < / a > < span class = "lineno" > 191< / span > std::vector< int> start,< / div >
< div class = "line" > < a id = "l00192" name = "l00192" > < / a > < span class = "lineno" > 192< / span > std::vector< int> stop,< / div >
< div class = "line" > < a id = "l00193" name = "l00193" > < / a > < span class = "lineno" > 193< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00194" name = "l00194" > < / a > < span class = "lineno" > 194< / span > < / div >
< div class = "line" > < a id = "l00196" name = "l00196" > < / a > < span class = "lineno" > 196< / span > std::vector< array> < / div >
< div class = "line" > < a id = "l00197" name = "l00197" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga7534290bceab5fb3831a05d67bebce7d" > 197< / 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 = "l00198" name = "l00198" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga56882d24e5fde59c266774624c892d41" > 198< / 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 = "l00199" name = "l00199" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga2cfcb1a53924882e30476c9016c5de74" > 199< / a > < / span > std::vector< array> < a class = "code hl_function" href = "group__ops.html#ga7534290bceab5fb3831a05d67bebce7d" > split< / a > (< / div >
< div class = "line" > < a id = "l00200" name = "l00200" > < / a > < span class = "lineno" > 200< / 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 = "l00201" name = "l00201" > < / a > < span class = "lineno" > 201< / span > < span class = "keyword" > const< / span > std::vector< int> & indices,< / div >
< div class = "line" > < a id = "l00202" name = "l00202" > < / a > < span class = "lineno" > 202< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l00203" name = "l00203" > < / a > < span class = "lineno" > 203< / span > < 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 > std::vector< array> < / div >
< div class = "line" > < a id = "l00205" name = "l00205" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gac324dfa3e26d3a14a35ab7962e36f0e1" > 205< / 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 = "l00206" name = "l00206" > < / a > < span class = "lineno" > 206< / span > < / div >
< div class = "line" > < a id = "l00208" name = "l00208" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga577c911618575314de63d1060656a26e" > 208< / a > < / span > std::vector< array> < a class = "code hl_function" href = "group__ops.html#ga577c911618575314de63d1060656a26e" > meshgrid< / a > (< / div >
< div class = "line" > < a id = "l00209" name = "l00209" > < / a > < span class = "lineno" > 209< / span > < span class = "keyword" > const< / span > std::vector< array> & arrays,< / div >
< div class = "line" > < a id = "l00210" name = "l00210" > < / a > < span class = "lineno" > 210< / span > < span class = "keywordtype" > bool< / span > sparse = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00211" name = "l00211" > < / a > < span class = "lineno" > 211< / span > std::string indexing = < span class = "stringliteral" > " xy" < / span > ,< / div >
< div class = "line" > < a id = "l00212" name = "l00212" > < / a > < span class = "lineno" > 212< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00213" name = "l00213" > < / a > < span class = "lineno" > 213< / span > < / div >
< div class = "line" > < a id = "l00217" name = "l00217" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga157cd7c23f9b306fee2e1eb2b9bf1dd8" > 217< / 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 = "l00218" name = "l00218" > < / a > < span class = "lineno" > 218< / 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 = "l00219" name = "l00219" > < / a > < span class = "lineno" > 219< / span > < span class = "keyword" > const< / span > std::optional< array> & a_min = std::nullopt,< / div >
< div class = "line" > < a id = "l00220" name = "l00220" > < / a > < span class = "lineno" > 220< / span > < span class = "keyword" > const< / span > std::optional< array> & a_max = std::nullopt,< / div >
2024-06-07 11:28:06 +08:00
< div class = "line" > < a id = "l00221" name = "l00221" > < / a > < span class = "lineno" > 221< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00222" name = "l00222" > < / a > < span class = "lineno" > 222< / span > < / div >
< div class = "line" > < a id = "l00224" name = "l00224" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gabdc36fa65697d0361c8d67495de77129" > 224< / 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 = "l00225" name = "l00225" > < / a > < span class = "lineno" > 225< / span > < span class = "keyword" > const< / span > std::vector< array> & arrays,< / div >
< div class = "line" > < a id = "l00226" name = "l00226" > < / a > < span class = "lineno" > 226< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l00227" name = "l00227" > < / a > < span class = "lineno" > 227< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00228" name = "l00228" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaa95c34ca3a8877f2c50cb60e7fa312b8" > 228< / 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 = "l00229" name = "l00229" > < / a > < span class = "lineno" > 229< / span > < / div >
< div class = "line" > < a id = "l00231" name = "l00231" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaf8f2ec2b98a4b59eca73d7471df6e032" > 231< / 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 = "l00232" name = "l00232" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga82216209dce901296fc737fe8efa5c94" > 232< / 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 >
2024-06-07 11:28:06 +08:00
< div class = "line" > < a id = "l00233" name = "l00233" > < / a > < span class = "lineno" > 233< / span > < / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00235" name = "l00235" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gab49e3a687e826554ed1574186e8ae974" > 235< / 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 = "l00236" name = "l00236" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga4f75f5d5db999f02f43ecbc6dccf3ba6" > 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#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 = "l00237" name = "l00237" > < / a > < span class = "lineno" > 237< / span > < / div >
< div class = "line" > < a id = "l00238" name = "l00238" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gab105a57b9a4d84496fe1e4d60e13d361" > 238< / 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 = "l00239" name = "l00239" > < / a > < span class = "lineno" > 239< / span > < / div >
< div class = "line" > < a id = "l00241" name = "l00241" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b" > 241< / 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 = "foldopen00242" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00242" name = "l00242" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga260ac332956f3a6bf1dfdb9095c84dc5" > 242< / 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 = "l00243" name = "l00243" > < / a > < span class = "lineno" > 243< / 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 = "l00244" name = "l00244" > < / a > < span class = "lineno" > 244< / span > std::initializer_list< int> axes,< / div >
< div class = "line" > < a id = "l00245" name = "l00245" > < / a > < span class = "lineno" > 245< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {}) {< / div >
< div class = "line" > < a id = "l00246" name = "l00246" > < / a > < span class = "lineno" > 246< / 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 = "l00247" name = "l00247" > < / a > < span class = "lineno" > 247< / span > }< / div >
< / div >
< div class = "line" > < a id = "l00248" name = "l00248" > < / a > < span class = "lineno" > 248< / span > < / div >
< div class = "line" > < a id = "l00250" name = "l00250" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gabc46eed81ab6c6247903e4ec0c4ec1fb" > 250< / 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 = "l00251" name = "l00251" > < / a > < span class = "lineno" > 251< / span > < / div >
< div class = "line" > < a id = "l00253" name = "l00253" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga24067d10a842db2c9d509ea48135a2c3" > 253< / 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 = "l00254" name = "l00254" > < / a > < span class = "lineno" > 254< / 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 = "l00255" name = "l00255" > < / a > < span class = "lineno" > 255< / span > < span class = "keywordtype" > int< / span > source,< / div >
< div class = "line" > < a id = "l00256" name = "l00256" > < / a > < span class = "lineno" > 256< / span > < span class = "keywordtype" > int< / span > destination,< / div >
< div class = "line" > < a id = "l00257" name = "l00257" > < / a > < span class = "lineno" > 257< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00258" name = "l00258" > < / a > < span class = "lineno" > 258< / span > < / div >
< div class = "line" > < a id = "l00260" name = "l00260" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga4fd04d6472ae7cdb94a174edfe338ea3" > 260< / 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#ga4fd04d6472ae7cdb94a174edfe338ea3" > pad< / a > (< / div >
< div class = "line" > < a id = "l00261" name = "l00261" > < / a > < span class = "lineno" > 261< / 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 = "l00262" name = "l00262" > < / a > < span class = "lineno" > 262< / span > < span class = "keyword" > const< / span > std::vector< int> & axes,< / div >
< div class = "line" > < a id = "l00263" name = "l00263" > < / a > < span class = "lineno" > 263< / span > < span class = "keyword" > const< / span > std::vector< int> & low_pad_size,< / div >
< div class = "line" > < a id = "l00264" name = "l00264" > < / a > < span class = "lineno" > 264< / span > < span class = "keyword" > const< / span > std::vector< int> & high_pad_size,< / div >
< div class = "line" > < a id = "l00265" name = "l00265" > < / a > < span class = "lineno" > 265< / 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 = "l00266" name = "l00266" > < / a > < span class = "lineno" > 266< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00267" name = "l00267" > < / a > < span class = "lineno" > 267< / span > < / div >
< div class = "line" > < a id = "l00269" name = "l00269" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaca4e00d5e4ed9e9f29d56a17f19c2e31" > 269< / 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#ga4fd04d6472ae7cdb94a174edfe338ea3" > pad< / a > (< / div >
< div class = "line" > < a id = "l00270" name = "l00270" > < / a > < span class = "lineno" > 270< / 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 = "l00271" name = "l00271" > < / a > < span class = "lineno" > 271< / span > < span class = "keyword" > const< / span > std::vector< std::pair< int, int> > & pad_width,< / div >
< div class = "line" > < a id = "l00272" name = "l00272" > < / a > < span class = "lineno" > 272< / 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 = "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#ga608164c4f5b4ac2fba374219285d90bc" > 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#ga4fd04d6472ae7cdb94a174edfe338ea3" > 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 > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00279" name = "l00279" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga2e57e26c4d7f3d18802c164656cf915c" > 279< / 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#ga4fd04d6472ae7cdb94a174edfe338ea3" > pad< / a > (< / div >
< div class = "line" > < a id = "l00280" name = "l00280" > < / a > < span class = "lineno" > 280< / 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 = "l00281" name = "l00281" > < / a > < span class = "lineno" > 281< / span > < span class = "keywordtype" > int< / span > pad_width,< / div >
< div class = "line" > < a id = "l00282" name = "l00282" > < / a > < span class = "lineno" > 282< / 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 = "l00283" name = "l00283" > < / a > < span class = "lineno" > 283< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00284" name = "l00284" > < / a > < span class = "lineno" > 284< / span > < / div >
< div class = "line" > < a id = "l00286" name = "l00286" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga68da0176fefbe0c0096783c6fd926c6a" > 286< / 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 >
2024-06-07 11:28:06 +08:00
< div class = "line" > < a id = "l00287" name = "l00287" > < / a > < span class = "lineno" > 287< / span > < / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00289" name = "l00289" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gad256e86cc1a6e6b3832e392baa90318d" > 289< / 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 = "l00290" name = "l00290" > < / a > < span class = "lineno" > 290< / 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 = "l00291" name = "l00291" > < / a > < span class = "lineno" > 291< / span > < span class = "keyword" > const< / span > std::vector< int> & shape,< / div >
< div class = "line" > < a id = "l00292" name = "l00292" > < / a > < span class = "lineno" > 292< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00293" name = "l00293" > < / a > < span class = "lineno" > 293< / span > < / div >
< div class = "line" > < a id = "l00295" name = "l00295" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gab783890428b596f715dc7dd2057eae99" > 295< / a > < / span > std::vector< array> < a class = "code hl_function" href = "group__ops.html#gab783890428b596f715dc7dd2057eae99" > broadcast_arrays< / a > (< / div >
< div class = "line" > < a id = "l00296" name = "l00296" > < / a > < span class = "lineno" > 296< / span > < span class = "keyword" > const< / span > std::vector< array> & inputs,< / div >
< div class = "line" > < a id = "l00297" name = "l00297" > < / a > < span class = "lineno" > 297< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00298" name = "l00298" > < / a > < span class = "lineno" > 298< / span > < / div >
< div class = "line" > < a id = "l00300" name = "l00300" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga33638dc3a9972dd02be12d0eb85f9bde" > 300< / 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 = "foldopen00301" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00301" name = "l00301" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaa30cf69f3d22f65615f5e1696dd5703f" > 301< / 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 = "l00302" name = "l00302" > < / a > < span class = "lineno" > 302< / 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 = "l00303" name = "l00303" > < / a > < span class = "lineno" > 303< / span > }< / div >
< / div >
< div class = "line" > < a id = "l00304" name = "l00304" > < / a > < span class = "lineno" > 304< / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "foldopen" id = "foldopen00305" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00305" name = "l00305" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaf115782d009ac2a547fcca395c9ec797" > 305< / 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 = "l00306" name = "l00306" > < / a > < span class = "lineno" > 306< / 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 = "l00307" name = "l00307" > < / a > < span class = "lineno" > 307< / span > }< / div >
< / div >
< div class = "line" > < a id = "l00308" name = "l00308" > < / a > < span class = "lineno" > 308< / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
2024-06-07 11:28:06 +08:00
< div class = "foldopen" id = "foldopen00309" data-start = "{" data-end = "}" >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00309" name = "l00309" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga3ad3ed7aece2650943a35082dbe3a0a5" > 309< / 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 = "l00310" name = "l00310" > < / a > < span class = "lineno" > 310< / 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 >
2024-06-07 11:28:06 +08:00
< div class = "line" > < a id = "l00311" name = "l00311" > < / a > < span class = "lineno" > 311< / span > }< / div >
< / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00312" name = "l00312" > < / a > < span class = "lineno" > 312< / span > < / div >
< div class = "line" > < a id = "l00314" name = "l00314" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga28f22c5d2c399eee53be7b3facc11103" > 314< / 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 = "foldopen00315" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00315" name = "l00315" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga0ac483d85f23252ca8757e9926d5a3c5" > 315< / 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 = "l00316" name = "l00316" > < / a > < span class = "lineno" > 316< / 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 = "l00317" name = "l00317" > < / a > < span class = "lineno" > 317< / span > }< / div >
2024-06-07 11:28:06 +08:00
< / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00318" name = "l00318" > < / a > < span class = "lineno" > 318< / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "foldopen" id = "foldopen00319" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00319" name = "l00319" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga3fecba9f3cb9a19afd8ca492cf509ce0" > 319< / 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 = "l00320" name = "l00320" > < / a > < span class = "lineno" > 320< / 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 = "l00321" name = "l00321" > < / a > < span class = "lineno" > 321< / span > }< / div >
2024-06-07 11:28:06 +08:00
< / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00322" name = "l00322" > < / a > < span class = "lineno" > 322< / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
2024-06-07 11:28:06 +08:00
< div class = "foldopen" id = "foldopen00323" data-start = "{" data-end = "}" >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00323" name = "l00323" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaebbf1cfde388c7480159a03c92c9a385" > 323< / 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 = "l00324" name = "l00324" > < / a > < span class = "lineno" > 324< / 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 >
2024-06-07 11:28:06 +08:00
< div class = "line" > < a id = "l00325" name = "l00325" > < / a > < span class = "lineno" > 325< / span > }< / div >
< / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00326" name = "l00326" > < / a > < span class = "lineno" > 326< / span > < / div >
< div class = "line" > < a id = "l00328" name = "l00328" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaf4ec7bfc1ad13b891f1f3ef1772ef04d" > 328< / 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 = "foldopen00329" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00329" name = "l00329" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga74fd2777adef10e6fe628a9cdadb01cb" > 329< / 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 = "l00330" name = "l00330" > < / a > < span class = "lineno" > 330< / 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 = "l00331" name = "l00331" > < / a > < span class = "lineno" > 331< / span > }< / div >
2024-06-07 11:28:06 +08:00
< / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00332" name = "l00332" > < / a > < span class = "lineno" > 332< / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "foldopen" id = "foldopen00333" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00333" name = "l00333" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga32e106e794e2c32e4e7decee2df2477f" > 333< / 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 = "l00334" name = "l00334" > < / a > < span class = "lineno" > 334< / 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 = "l00335" name = "l00335" > < / a > < span class = "lineno" > 335< / span > }< / div >
2024-06-07 11:28:06 +08:00
< / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00336" name = "l00336" > < / a > < span class = "lineno" > 336< / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
2024-06-07 11:28:06 +08:00
< div class = "foldopen" id = "foldopen00337" data-start = "{" data-end = "}" >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00337" name = "l00337" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga96552b90e89923c5d2064cc427775ec5" > 337< / 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 = "l00338" name = "l00338" > < / a > < span class = "lineno" > 338< / 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 >
2024-06-07 11:28:06 +08:00
< div class = "line" > < a id = "l00339" name = "l00339" > < / a > < span class = "lineno" > 339< / span > }< / div >
< / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00340" name = "l00340" > < / a > < span class = "lineno" > 340< / span > < / div >
< div class = "line" > < a id = "l00342" name = "l00342" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga7153071bcfff6faad21332163fb9a430" > 342< / 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 = "foldopen00343" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00343" name = "l00343" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga3a41895f25ed083a36994d95fa102546" > 343< / 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 = "l00344" name = "l00344" > < / a > < span class = "lineno" > 344< / 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 = "l00345" name = "l00345" > < / a > < span class = "lineno" > 345< / span > }< / div >
2024-06-07 11:28:06 +08:00
< / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00346" name = "l00346" > < / a > < span class = "lineno" > 346< / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "foldopen" id = "foldopen00347" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00347" name = "l00347" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaf509f2cb3b18963232f20d6c3bd229b2" > 347< / 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 = "l00348" name = "l00348" > < / a > < span class = "lineno" > 348< / 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 = "l00349" name = "l00349" > < / a > < span class = "lineno" > 349< / span > }< / div >
2024-06-07 11:28:06 +08:00
< / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00350" name = "l00350" > < / a > < span class = "lineno" > 350< / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
2024-06-07 11:28:06 +08:00
< div class = "foldopen" id = "foldopen00351" data-start = "{" data-end = "}" >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00351" name = "l00351" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gafa0eb25d5978674bfc9e59d4145ec590" > 351< / 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 = "l00352" name = "l00352" > < / a > < span class = "lineno" > 352< / 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 >
2024-06-07 11:28:06 +08:00
< div class = "line" > < a id = "l00353" name = "l00353" > < / a > < span class = "lineno" > 353< / span > }< / div >
< / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00354" name = "l00354" > < / a > < span class = "lineno" > 354< / span > < / div >
< div class = "line" > < a id = "l00356" name = "l00356" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga9142b8d717699a8abfa2a7398891ff8a" > 356< / 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 = "foldopen00357" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00357" name = "l00357" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaee41e2b8f61d563200ff03575ac1d6c3" > 357< / 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 = "l00358" name = "l00358" > < / a > < span class = "lineno" > 358< / 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 = "l00359" name = "l00359" > < / a > < span class = "lineno" > 359< / span > }< / div >
2024-06-07 11:28:06 +08:00
< / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00360" name = "l00360" > < / a > < span class = "lineno" > 360< / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "foldopen" id = "foldopen00361" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00361" name = "l00361" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga1ef8ea11cf15ce628c54201fa42748ef" > 361< / 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 = "l00362" name = "l00362" > < / a > < span class = "lineno" > 362< / 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 = "l00363" name = "l00363" > < / a > < span class = "lineno" > 363< / span > }< / div >
2024-06-07 11:28:06 +08:00
< / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00364" name = "l00364" > < / a > < span class = "lineno" > 364< / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
2024-06-07 11:28:06 +08:00
< div class = "foldopen" id = "foldopen00365" data-start = "{" data-end = "}" >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00365" name = "l00365" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga95e72226dc7a79c40b3d16f990922050" > 365< / 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 = "l00366" name = "l00366" > < / a > < span class = "lineno" > 366< / 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 >
2024-06-07 11:28:06 +08:00
< div class = "line" > < a id = "l00367" name = "l00367" > < / a > < span class = "lineno" > 367< / span > }< / div >
< / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00368" name = "l00368" > < / a > < span class = "lineno" > 368< / span > < / div >
< div class = "line" > < a id = "l00370" name = "l00370" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga0d49e0c7011d0573c369c13c8f045a09" > 370< / 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 = "foldopen00371" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00371" name = "l00371" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga4c8b8a1632944acaae50f0de6c23ece6" > 371< / 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 = "l00372" name = "l00372" > < / a > < span class = "lineno" > 372< / 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 = "l00373" name = "l00373" > < / a > < span class = "lineno" > 373< / span > }< / div >
< / div >
< div class = "line" > < a id = "l00374" name = "l00374" > < / a > < span class = "lineno" > 374< / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "foldopen" id = "foldopen00375" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00375" name = "l00375" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga150a9be467c9f91482a6d6fc13504bc4" > 375< / 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 = "l00376" name = "l00376" > < / a > < span class = "lineno" > 376< / 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 = "l00377" name = "l00377" > < / a > < span class = "lineno" > 377< / span > }< / div >
< / div >
< div class = "line" > < a id = "l00378" name = "l00378" > < / a > < span class = "lineno" > 378< / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "foldopen" id = "foldopen00379" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00379" name = "l00379" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga624eeccef0cc4b130e1325abfea057cb" > 379< / 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 = "l00380" name = "l00380" > < / a > < span class = "lineno" > 380< / 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 = "l00381" name = "l00381" > < / a > < span class = "lineno" > 381< / span > }< / div >
< / div >
< div class = "line" > < a id = "l00382" name = "l00382" > < / a > < span class = "lineno" > 382< / span > < / div >
< div class = "line" > < a id = "l00384" name = "l00384" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga8f3059336ee0c87207b1f8c6ab312645" > 384< / 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 = "l00385" name = "l00385" > < / a > < span class = "lineno" > 385< / 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 = "l00386" name = "l00386" > < / a > < span class = "lineno" > 386< / 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 = "l00387" name = "l00387" > < / a > < span class = "lineno" > 387< / span > < span class = "keywordtype" > bool< / span > equal_nan,< / div >
< div class = "line" > < a id = "l00388" name = "l00388" > < / a > < span class = "lineno" > 388< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00389" name = "l00389" > < / a > < span class = "lineno" > 389< / span > < span class = "keyword" > inline< / span > array< / div >
< div class = "foldopen" id = "foldopen00390" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00390" name = "l00390" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaf79cf0271ca0105d7b14295a90d0ed14" > 390< / 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 = "l00391" name = "l00391" > < / a > < span class = "lineno" > 391< / 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 = "l00392" name = "l00392" > < / a > < span class = "lineno" > 392< / span > }< / div >
< / div >
2024-06-07 11:28:06 +08:00
< div class = "line" > < a id = "l00393" name = "l00393" > < / a > < span class = "lineno" > 393< / span > < / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00394" name = "l00394" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga175592792471b0ffb45196dca4711ba6" > 394< / 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 >
2024-06-07 11:28:06 +08:00
< div class = "line" > < a id = "l00395" name = "l00395" > < / a > < span class = "lineno" > 395< / span > < / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00396" name = "l00396" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga8fc238d5e5d1153e69da8b36015d9844" > 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#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 = "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#gad80f7c4a58c12b6cb30a8b9a73008993" > 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#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 = "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#ga1940523da381ed7be50656a3bc465ff3" > 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#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 = "l00401" name = "l00401" > < / a > < span class = "lineno" > 401< / span > < / div >
< div class = "line" > < a id = "l00403" name = "l00403" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga8a2056f8c9bb30914c40bcf509386491" > 403< / 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 = "l00404" name = "l00404" > < / a > < span class = "lineno" > 404< / 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 = "l00405" name = "l00405" > < / a > < span class = "lineno" > 405< / 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 = "l00406" name = "l00406" > < / a > < span class = "lineno" > 406< / 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 = "l00407" name = "l00407" > < / a > < span class = "lineno" > 407< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
2024-06-07 11:28:06 +08:00
< div class = "line" > < a id = "l00408" name = "l00408" > < / a > < span class = "lineno" > 408< / span > < / div >
2024-07-26 02:59:11 +08:00
< div class = "line" > < a id = "l00410" name = "l00410" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga506d1b63e391e45a7d8dc10cfd71e78d" > 410< / 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#ga506d1b63e391e45a7d8dc10cfd71e78d" > nan_to_num< / a > (< / div >
< div class = "line" > < a id = "l00411" name = "l00411" > < / a > < span class = "lineno" > 411< / 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 = "l00412" name = "l00412" > < / a > < span class = "lineno" > 412< / span > < span class = "keywordtype" > float< / span > nan = 0.0f,< / div >
< div class = "line" > < a id = "l00413" name = "l00413" > < / a > < span class = "lineno" > 413< / span > < span class = "keyword" > const< / span > std::optional< float> & posinf = std::nullopt,< / div >
< div class = "line" > < a id = "l00414" name = "l00414" > < / a > < span class = "lineno" > 414< / span > < span class = "keyword" > const< / span > std::optional< float> & neginf = std::nullopt,< / div >
< div class = "line" > < a id = "l00415" name = "l00415" > < / a > < span class = "lineno" > 415< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00416" name = "l00416" > < / a > < span class = "lineno" > 416< / span > < / div >
< div class = "line" > < a id = "l00418" name = "l00418" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga3b1b90ef1275ca17655b6d7f25d3ee68" > 418< / 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 = "foldopen00419" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00419" name = "l00419" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga3689e12e8f42dadb4cbe2b07dc4099f4" > 419< / 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 = "l00420" name = "l00420" > < / a > < span class = "lineno" > 420< / 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 = "l00421" name = "l00421" > < / a > < span class = "lineno" > 421< / span > }< / div >
< / div >
< div class = "line" > < a id = "l00422" name = "l00422" > < / a > < span class = "lineno" > 422< / span > < / div >
< div class = "line" > < a id = "l00424" name = "l00424" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaf0cd4257de7542daf9faf5e605e31020" > 424< / 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 = "l00425" name = "l00425" > < / a > < span class = "lineno" > 425< / 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 = "l00426" name = "l00426" > < / a > < span class = "lineno" > 426< / 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 = "l00427" name = "l00427" > < / a > < span class = "lineno" > 427< / span > < span class = "keywordtype" > double< / span > rtol = 1e-5,< / div >
< div class = "line" > < a id = "l00428" name = "l00428" > < / a > < span class = "lineno" > 428< / span > < span class = "keywordtype" > double< / span > atol = 1e-8,< / div >
< div class = "line" > < a id = "l00429" name = "l00429" > < / a > < span class = "lineno" > 429< / span > < span class = "keywordtype" > bool< / span > equal_nan = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00430" name = "l00430" > < / a > < span class = "lineno" > 430< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00431" name = "l00431" > < / a > < span class = "lineno" > 431< / span > < / div >
< div class = "line" > < a id = "l00434" name = "l00434" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga51eac95c04400921c54716de14b52491" > 434< / 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 = "l00435" name = "l00435" > < / a > < span class = "lineno" > 435< / 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 = "l00436" name = "l00436" > < / a > < span class = "lineno" > 436< / 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 = "l00437" name = "l00437" > < / a > < span class = "lineno" > 437< / span > < span class = "keywordtype" > double< / span > rtol = 1e-5,< / div >
< div class = "line" > < a id = "l00438" name = "l00438" > < / a > < span class = "lineno" > 438< / span > < span class = "keywordtype" > double< / span > atol = 1e-8,< / div >
< div class = "line" > < a id = "l00439" name = "l00439" > < / a > < span class = "lineno" > 439< / span > < span class = "keywordtype" > bool< / span > equal_nan = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00440" name = "l00440" > < / a > < span class = "lineno" > 440< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00441" name = "l00441" > < / a > < span class = "lineno" > 441< / span > < / div >
< div class = "line" > < a id = "l00446" name = "l00446" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gac0919c6ba53aea35a7683dea7e9a9a59" > 446< / 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 = "l00447" name = "l00447" > < / a > < span class = "lineno" > 447< / 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 = "l00448" name = "l00448" > < / a > < span class = "lineno" > 448< / span > < span class = "keyword" > const< / span > std::vector< int> & axes,< / div >
< div class = "line" > < a id = "l00449" name = "l00449" > < / a > < span class = "lineno" > 449< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00450" name = "l00450" > < / a > < span class = "lineno" > 450< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00451" name = "l00451" > < / a > < span class = "lineno" > 451< / span > < / div >
< div class = "line" > < a id = "l00456" name = "l00456" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gae2d5fcc5b62d673cca76c08b7b4afbbc" > 456< / 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 = "l00457" name = "l00457" > < / a > < span class = "lineno" > 457< / 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 = "l00458" name = "l00458" > < / a > < span class = "lineno" > 458< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l00459" name = "l00459" > < / a > < span class = "lineno" > 459< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00460" name = "l00460" > < / a > < span class = "lineno" > 460< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00461" name = "l00461" > < / a > < span class = "lineno" > 461< / span > < / div >
< div class = "line" > < a id = "l00463" name = "l00463" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga8598dd718fb05cb28535e250372d4e6f" > 463< / 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 = "foldopen00464" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00464" name = "l00464" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gad37df97f253a963bece124198dbaf9ba" > 464< / 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 = "l00465" name = "l00465" > < / a > < span class = "lineno" > 465< / 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 = "l00466" name = "l00466" > < / a > < span class = "lineno" > 466< / span > }< / div >
< / div >
< div class = "line" > < a id = "l00467" name = "l00467" > < / a > < span class = "lineno" > 467< / span > < / div >
< div class = "line" > < a id = "l00472" name = "l00472" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaf240618fc8b06debf5f56e97e84f18ef" > 472< / 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 = "l00473" name = "l00473" > < / a > < span class = "lineno" > 473< / 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 = "l00474" name = "l00474" > < / a > < span class = "lineno" > 474< / span > < span class = "keyword" > const< / span > std::vector< int> & axes,< / div >
< div class = "line" > < a id = "l00475" name = "l00475" > < / a > < span class = "lineno" > 475< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00476" name = "l00476" > < / a > < span class = "lineno" > 476< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00477" name = "l00477" > < / a > < span class = "lineno" > 477< / span > < / div >
< div class = "line" > < a id = "l00482" name = "l00482" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gab1d56277d468a55227f4dad6bc2fc1ce" > 482< / 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 = "l00483" name = "l00483" > < / a > < span class = "lineno" > 483< / 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 = "l00484" name = "l00484" > < / a > < span class = "lineno" > 484< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l00485" name = "l00485" > < / a > < span class = "lineno" > 485< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00486" name = "l00486" > < / a > < span class = "lineno" > 486< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00487" name = "l00487" > < / a > < span class = "lineno" > 487< / span > < / div >
< div class = "line" > < a id = "l00489" name = "l00489" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gade905ee92eb6ab7edfc312aeddfbaeb6" > 489< / 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 = "foldopen00490" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00490" name = "l00490" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga3627754d7868487bdab1bd83f05d9c81" > 490< / 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 = "l00491" name = "l00491" > < / a > < span class = "lineno" > 491< / 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 = "l00492" name = "l00492" > < / a > < span class = "lineno" > 492< / span > }< / div >
< / div >
< div class = "line" > < a id = "l00493" name = "l00493" > < / a > < span class = "lineno" > 493< / span > < / div >
< div class = "line" > < a id = "l00495" name = "l00495" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaccd0a6be2c5b5128fdc2d87b5c8e67f4" > 495< / 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 = "l00496" name = "l00496" > < / a > < span class = "lineno" > 496< / 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 = "l00497" name = "l00497" > < / a > < span class = "lineno" > 497< / span > < span class = "keyword" > const< / span > std::vector< int> & axes,< / div >
< div class = "line" > < a id = "l00498" name = "l00498" > < / a > < span class = "lineno" > 498< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00499" name = "l00499" > < / a > < span class = "lineno" > 499< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00500" name = "l00500" > < / a > < span class = "lineno" > 500< / span > < / div >
< div class = "line" > < a id = "l00502" name = "l00502" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gafcd39b0bf39a56c26a967981c7ab8a8d" > 502< / 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 = "l00503" name = "l00503" > < / a > < span class = "lineno" > 503< / 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 = "l00504" name = "l00504" > < / a > < span class = "lineno" > 504< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l00505" name = "l00505" > < / a > < span class = "lineno" > 505< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00506" name = "l00506" > < / a > < span class = "lineno" > 506< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00507" name = "l00507" > < / a > < span class = "lineno" > 507< / span > < / div >
< div class = "line" > < a id = "l00509" name = "l00509" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gade46e768fd46b8b640eb16f26abeecef" > 509< / 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 = "foldopen00510" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00510" name = "l00510" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga52b59fdd8e8430538e564f5bbcfa31e6" > 510< / 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 = "l00511" name = "l00511" > < / a > < span class = "lineno" > 511< / 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 = "l00512" name = "l00512" > < / a > < span class = "lineno" > 512< / span > }< / div >
< / div >
< div class = "line" > < a id = "l00513" name = "l00513" > < / a > < span class = "lineno" > 513< / span > < / div >
< div class = "line" > < a id = "l00515" name = "l00515" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga066161f3d3e395a1d76c638cb680d444" > 515< / 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 = "l00516" name = "l00516" > < / a > < span class = "lineno" > 516< / 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 = "l00517" name = "l00517" > < / a > < span class = "lineno" > 517< / span > < span class = "keyword" > const< / span > std::vector< int> & axes,< / div >
< div class = "line" > < a id = "l00518" name = "l00518" > < / a > < span class = "lineno" > 518< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00519" name = "l00519" > < / a > < span class = "lineno" > 519< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00520" name = "l00520" > < / a > < span class = "lineno" > 520< / span > < / div >
< div class = "line" > < a id = "l00522" name = "l00522" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga45fba73eab0e3b6e128ed3ce2f43a5da" > 522< / 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 = "l00523" name = "l00523" > < / a > < span class = "lineno" > 523< / 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 = "l00524" name = "l00524" > < / a > < span class = "lineno" > 524< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l00525" name = "l00525" > < / a > < span class = "lineno" > 525< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00526" name = "l00526" > < / a > < span class = "lineno" > 526< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00527" name = "l00527" > < / a > < span class = "lineno" > 527< / span > < / div >
< div class = "line" > < a id = "l00529" name = "l00529" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga7e133df686439588a8cd1fb10ce0c6e9" > 529< / 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 = "foldopen00530" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00530" name = "l00530" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga7d7b38d118fa2613214078ef0f7d5a42" > 530< / 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 = "l00531" name = "l00531" > < / a > < span class = "lineno" > 531< / 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 = "l00532" name = "l00532" > < / a > < span class = "lineno" > 532< / span > }< / div >
< / div >
< div class = "line" > < a id = "l00533" name = "l00533" > < / a > < span class = "lineno" > 533< / span > < / div >
< div class = "line" > < a id = "l00536" name = "l00536" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga78ddeb966cbe7a5b0aa17e1de43025f2" > 536< / 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 = "l00537" name = "l00537" > < / a > < span class = "lineno" > 537< / 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 = "l00538" name = "l00538" > < / a > < span class = "lineno" > 538< / span > < span class = "keyword" > const< / span > std::vector< int> & axes,< / div >
< div class = "line" > < a id = "l00539" name = "l00539" > < / a > < span class = "lineno" > 539< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00540" name = "l00540" > < / a > < span class = "lineno" > 540< / span > < span class = "keywordtype" > int< / span > ddof = 0,< / div >
< div class = "line" > < a id = "l00541" name = "l00541" > < / a > < span class = "lineno" > 541< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00542" name = "l00542" > < / a > < span class = "lineno" > 542< / span > < / div >
< div class = "line" > < a id = "l00545" name = "l00545" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga4fbf3e3f98f2e4956faf87af320aa9d0" > 545< / 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 = "l00546" name = "l00546" > < / a > < span class = "lineno" > 546< / 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 = "l00547" name = "l00547" > < / a > < span class = "lineno" > 547< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l00548" name = "l00548" > < / a > < span class = "lineno" > 548< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00549" name = "l00549" > < / a > < span class = "lineno" > 549< / span > < span class = "keywordtype" > int< / span > ddof = 0,< / div >
< div class = "line" > < a id = "l00550" name = "l00550" > < / a > < span class = "lineno" > 550< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00551" name = "l00551" > < / a > < span class = "lineno" > 551< / span > < / div >
< div class = "line" > < a id = "l00553" name = "l00553" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga2a466024f8061febc0a64be557644cb0" > 553< / 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 = "foldopen00554" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00554" name = "l00554" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gafdcb04d77c64405a3990078a77dd984c" > 554< / 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 = "l00555" name = "l00555" > < / a > < span class = "lineno" > 555< / 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 = "l00556" name = "l00556" > < / a > < span class = "lineno" > 556< / span > }< / div >
< / div >
< div class = "line" > < a id = "l00557" name = "l00557" > < / a > < span class = "lineno" > 557< / span > < / div >
< div class = "line" > < a id = "l00560" name = "l00560" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga7f649970bf38b987b6ef847054f3c2f8" > 560< / 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 = "l00561" name = "l00561" > < / a > < span class = "lineno" > 561< / 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 = "l00562" name = "l00562" > < / a > < span class = "lineno" > 562< / span > < span class = "keyword" > const< / span > std::vector< int> & axes,< / div >
< div class = "line" > < a id = "l00563" name = "l00563" > < / a > < span class = "lineno" > 563< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00564" name = "l00564" > < / a > < span class = "lineno" > 564< / span > < span class = "keywordtype" > int< / span > ddof = 0,< / div >
< div class = "line" > < a id = "l00565" name = "l00565" > < / a > < span class = "lineno" > 565< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00566" name = "l00566" > < / a > < span class = "lineno" > 566< / span > < / div >
< div class = "line" > < a id = "l00569" name = "l00569" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga62721a206df671ef5797449eea97af9f" > 569< / 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 = "l00570" name = "l00570" > < / a > < span class = "lineno" > 570< / 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 = "l00571" name = "l00571" > < / a > < span class = "lineno" > 571< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l00572" name = "l00572" > < / a > < span class = "lineno" > 572< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00573" name = "l00573" > < / a > < span class = "lineno" > 573< / span > < span class = "keywordtype" > int< / span > ddof = 0,< / div >
< div class = "line" > < a id = "l00574" name = "l00574" > < / a > < span class = "lineno" > 574< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00575" name = "l00575" > < / a > < span class = "lineno" > 575< / span > < / div >
< div class = "line" > < a id = "l00577" name = "l00577" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga4a09b7241d564d92548bc2773eb1d544" > 577< / 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 = "foldopen00578" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00578" name = "l00578" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga61832191f3c42ea549cf04953edc3602" > 578< / 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 = "l00579" name = "l00579" > < / a > < span class = "lineno" > 579< / 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 = "l00580" name = "l00580" > < / a > < span class = "lineno" > 580< / span > }< / div >
< / div >
< div class = "line" > < a id = "l00581" name = "l00581" > < / a > < span class = "lineno" > 581< / span > < / div >
< div class = "line" > < a id = "l00583" name = "l00583" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga2b3935108f641e20a70dbf63f540d970" > 583< / 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 = "l00584" name = "l00584" > < / a > < span class = "lineno" > 584< / 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 = "l00585" name = "l00585" > < / a > < span class = "lineno" > 585< / span > < span class = "keyword" > const< / span > std::vector< int> & axes,< / div >
< div class = "line" > < a id = "l00586" name = "l00586" > < / a > < span class = "lineno" > 586< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00587" name = "l00587" > < / a > < span class = "lineno" > 587< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00588" name = "l00588" > < / a > < span class = "lineno" > 588< / span > < / div >
< div class = "line" > < a id = "l00590" name = "l00590" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga8a10a10b81c69996d0aca8ba401f8ff0" > 590< / 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 = "l00591" name = "l00591" > < / a > < span class = "lineno" > 591< / 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 = "l00592" name = "l00592" > < / a > < span class = "lineno" > 592< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l00593" name = "l00593" > < / a > < span class = "lineno" > 593< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00594" name = "l00594" > < / a > < span class = "lineno" > 594< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00595" name = "l00595" > < / a > < span class = "lineno" > 595< / span > < / div >
< div class = "line" > < a id = "l00597" name = "l00597" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga7fed87d96cc7741d8267f4eac83f5fe7" > 597< / 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 = "foldopen00598" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00598" name = "l00598" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga25be91d70a5f40341db0615a0b8bfedc" > 598< / 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 = "l00599" name = "l00599" > < / a > < span class = "lineno" > 599< / 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 = "l00600" name = "l00600" > < / a > < span class = "lineno" > 600< / span > }< / div >
< / div >
< div class = "line" > < a id = "l00601" name = "l00601" > < / a > < span class = "lineno" > 601< / span > < / div >
< div class = "line" > < a id = "l00603" name = "l00603" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga1ca7b6b91fe2459a7d83897bf013827f" > 603< / 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 = "l00604" name = "l00604" > < / a > < span class = "lineno" > 604< / 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 = "l00605" name = "l00605" > < / a > < span class = "lineno" > 605< / span > < span class = "keyword" > const< / span > std::vector< int> & axes,< / div >
< div class = "line" > < a id = "l00606" name = "l00606" > < / a > < span class = "lineno" > 606< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00607" name = "l00607" > < / a > < span class = "lineno" > 607< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00608" name = "l00608" > < / a > < span class = "lineno" > 608< / span > < / div >
< div class = "line" > < a id = "l00610" name = "l00610" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga7b638050e03a93f2896c981bc2850a47" > 610< / 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 = "l00611" name = "l00611" > < / a > < span class = "lineno" > 611< / 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 = "l00612" name = "l00612" > < / a > < span class = "lineno" > 612< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l00613" name = "l00613" > < / a > < span class = "lineno" > 613< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00614" name = "l00614" > < / a > < span class = "lineno" > 614< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00615" name = "l00615" > < / a > < span class = "lineno" > 615< / span > < / div >
< div class = "line" > < a id = "l00617" name = "l00617" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gab27599802617a4c8f9964ab5f4ffee12" > 617< / 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 = "foldopen00618" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00618" name = "l00618" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga0140b91e9cdfc3fef0da8e332f65a9e8" > 618< / 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 = "l00619" name = "l00619" > < / a > < span class = "lineno" > 619< / 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 = "l00620" name = "l00620" > < / a > < span class = "lineno" > 620< / span > }< / div >
< / div >
< div class = "line" > < a id = "l00621" name = "l00621" > < / a > < span class = "lineno" > 621< / span > < / div >
< div class = "line" > < a id = "l00623" name = "l00623" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga6efb83cd46436678c8f8c4af15cc00f5" > 623< / 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 = "l00624" name = "l00624" > < / a > < span class = "lineno" > 624< / 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 = "l00625" name = "l00625" > < / a > < span class = "lineno" > 625< / span > < span class = "keyword" > const< / span > std::vector< int> & axes,< / div >
< div class = "line" > < a id = "l00626" name = "l00626" > < / a > < span class = "lineno" > 626< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00627" name = "l00627" > < / a > < span class = "lineno" > 627< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00628" name = "l00628" > < / a > < span class = "lineno" > 628< / span > < / div >
< div class = "line" > < a id = "l00630" name = "l00630" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga36fa315eef677f4143868f552cd26d03" > 630< / 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 = "l00631" name = "l00631" > < / a > < span class = "lineno" > 631< / 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 = "l00632" name = "l00632" > < / a > < span class = "lineno" > 632< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l00633" name = "l00633" > < / a > < span class = "lineno" > 633< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00634" name = "l00634" > < / a > < span class = "lineno" > 634< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00635" name = "l00635" > < / a > < span class = "lineno" > 635< / span > < / div >
< div class = "line" > < a id = "l00637" name = "l00637" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga7c3bd5ef430a71dfd298e626741e3c71" > 637< / 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 = "foldopen00638" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00638" name = "l00638" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga6bc577c5ab10cd9c848ba81321595070" > 638< / 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 = "l00639" name = "l00639" > < / a > < span class = "lineno" > 639< / 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 = "l00640" name = "l00640" > < / a > < span class = "lineno" > 640< / span > }< / div >
< / div >
< div class = "line" > < a id = "l00641" name = "l00641" > < / a > < span class = "lineno" > 641< / span > < / div >
< div class = "line" > < a id = "l00643" name = "l00643" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaf66dc3c77b88e4009e0678eda41eca81" > 643< / 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 = "l00644" name = "l00644" > < / a > < span class = "lineno" > 644< / 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 = "l00645" name = "l00645" > < / a > < span class = "lineno" > 645< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l00646" name = "l00646" > < / a > < span class = "lineno" > 646< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00647" name = "l00647" > < / a > < span class = "lineno" > 647< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00648" name = "l00648" > < / a > < span class = "lineno" > 648< / span > < / div >
< div class = "line" > < a id = "l00650" name = "l00650" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gae60b0b5339b9c50b9970260faf613e83" > 650< / 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 = "foldopen00651" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00651" name = "l00651" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gae6f6c5a840320b336fdc9687e0ed56c8" > 651< / 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 = "l00652" name = "l00652" > < / a > < span class = "lineno" > 652< / 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 = "l00653" name = "l00653" > < / a > < span class = "lineno" > 653< / span > }< / div >
< / div >
< div class = "line" > < a id = "l00654" name = "l00654" > < / a > < span class = "lineno" > 654< / span > < / div >
< div class = "line" > < a id = "l00656" name = "l00656" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga2efa67466510fc26ab9ea8dff30f2ba5" > 656< / 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 = "l00657" name = "l00657" > < / a > < span class = "lineno" > 657< / 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 = "l00658" name = "l00658" > < / a > < span class = "lineno" > 658< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l00659" name = "l00659" > < / a > < span class = "lineno" > 659< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00660" name = "l00660" > < / a > < span class = "lineno" > 660< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00661" name = "l00661" > < / a > < span class = "lineno" > 661< / span > < / div >
< div class = "line" > < a id = "l00663" name = "l00663" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga7fb616054665b3c2d61fa234f501f079" > 663< / 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 = "l00664" name = "l00664" > < / a > < span class = "lineno" > 664< / span > < / div >
< div class = "line" > < a id = "l00666" name = "l00666" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaae1bc47aa737f705d0e5884270063fea" > 666< / 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 = "l00667" name = "l00667" > < / a > < span class = "lineno" > 667< / span > < / div >
< div class = "line" > < a id = "l00669" name = "l00669" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga8df3b2703bf671457422894dd870cdc5" > 669< / 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 = "l00670" name = "l00670" > < / a > < span class = "lineno" > 670< / span > < / div >
< div class = "line" > < a id = "l00672" name = "l00672" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga7878e0daa5a75f44e57b5fe948fa3ef6" > 672< / 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 = "l00673" name = "l00673" > < / a > < span class = "lineno" > 673< / span > < / div >
< div class = "line" > < a id = "l00678" name = "l00678" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gac1b30830a972fb9a2601379ad2b32405" > 678< / 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 = "l00679" name = "l00679" > < / a > < span class = "lineno" > 679< / span > < / div >
< div class = "line" > < a id = "l00684" name = "l00684" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga4fbea3a5f66cf81e3c119d1661119321" > 684< / 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 = "l00685" name = "l00685" > < / a > < span class = "lineno" > 685< / span > < / div >
< div class = "line" > < a id = "l00690" name = "l00690" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaf301c49c10fa9b95a9e8dc52ead1a8dd" > 690< / 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 = "l00691" name = "l00691" > < / a > < span class = "lineno" > 691< / span > < / div >
< div class = "line" > < a id = "l00696" name = "l00696" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga7b15c654c7463def57857a0e239989a3" > 696< / 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 = "l00697" name = "l00697" > < / a > < span class = "lineno" > 697< / span > < / div >
< div class = "line" > < a id = "l00699" name = "l00699" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga5487dd887c43e5341f3e68ffe47f0f5a" > 699< / 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 = "l00700" name = "l00700" > < / a > < span class = "lineno" > 700< / span > < / div >
< div class = "line" > < a id = "l00702" name = "l00702" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga35b8436c79ff953f6c809598b646f498" > 702< / 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 = "l00703" name = "l00703" > < / a > < span class = "lineno" > 703< / span > < / div >
< div class = "line" > < a id = "l00705" name = "l00705" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gacff4eb57c085d571e722083680267ac5" > 705< / 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 = "foldopen00706" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00706" name = "l00706" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga59be50b4e92f1dc20b53460cefa3910d" > 706< / 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 = "l00707" name = "l00707" > < / a > < span class = "lineno" > 707< / 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 = "l00708" name = "l00708" > < / a > < span class = "lineno" > 708< / span > }< / div >
< / div >
< div class = "line" > < a id = "l00709" name = "l00709" > < / a > < span class = "lineno" > 709< / span > < / div >
< div class = "line" > < a id = "l00711" name = "l00711" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gae3969c7bd24c4f3ab97831df28239689" > 711< / 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 = "l00712" name = "l00712" > < / a > < span class = "lineno" > 712< / 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 = "l00713" name = "l00713" > < / a > < span class = "lineno" > 713< / span > < span class = "keyword" > const< / span > std::vector< int> & axes,< / div >
< div class = "line" > < a id = "l00714" name = "l00714" > < / a > < span class = "lineno" > 714< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00715" name = "l00715" > < / a > < span class = "lineno" > 715< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00716" name = "l00716" > < / a > < span class = "lineno" > 716< / span > < / div >
< div class = "line" > < a id = "l00718" name = "l00718" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gafef5cb2159c16a60a95470cc823bdd44" > 718< / 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 = "l00719" name = "l00719" > < / a > < span class = "lineno" > 719< / 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 = "l00720" name = "l00720" > < / a > < span class = "lineno" > 720< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l00721" name = "l00721" > < / a > < span class = "lineno" > 721< / span > < span class = "keywordtype" > bool< / span > keepdims = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l00722" name = "l00722" > < / a > < span class = "lineno" > 722< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00723" name = "l00723" > < / a > < span class = "lineno" > 723< / span > < / div >
< div class = "line" > < a id = "l00725" name = "l00725" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga5528e80f5e8bad71e106a0cf9edd8920" > 725< / 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 = "l00726" name = "l00726" > < / a > < span class = "lineno" > 726< / span > < / div >
< div class = "line" > < a id = "l00728" name = "l00728" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga95d9a9425533b5ed1707eb00184dffc6" > 728< / 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 = "l00729" name = "l00729" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gade2eea48989f4caaf36e89f7bd2a8816" > 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#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 = "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#ga20f1a1a8c0cd6206485f9363f3915faa" > 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#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 = "l00733" name = "l00733" > < / a > < span class = "lineno" > 733< / span > < / div >
< div class = "line" > < a id = "l00735" name = "l00735" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gabca78d34ce93f0de2814e62225bb2a53" > 735< / 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 >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00736" name = "l00736" > < / a > < span class = "lineno" > 736< / span > < / div >
2024-07-26 02:59:11 +08:00
< div class = "line" > < a id = "l00738" name = "l00738" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga768977cda8d68cf23f464a6af9907876" > 738< / 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 = "l00739" name = "l00739" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaee1d774bb0843601d7a0a4257d616ae3" > 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#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 = "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#ga23768728e4dd070c917fbb0ed0d0c2ec" > 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#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 = "l00743" name = "l00743" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga27af56a98270d4d76d139f0f9171b83a" > 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#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 = "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#ga4d29556bb93e2f66916116cf1f062b36" > 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#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 = "l00747" name = "l00747" > < / a > < span class = "lineno" > 747< / span > < / div >
< div class = "line" > < a id = "l00749" name = "l00749" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga2d32d67cfd76785a72c43d89b94dc7d7" > 749< / 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 = "l00750" name = "l00750" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga26e5a043eaaaf066d1400adac9c11d0c" > 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#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 = "l00751" name = "l00751" > < / a > < span class = "lineno" > 751< / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "foldopen" id = "foldopen00752" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00752" name = "l00752" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga7d0ec8d01e7cefa6a6b25f11876761b5" > 752< / 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 = "l00753" name = "l00753" > < / a > < span class = "lineno" > 753< / 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 = "l00754" name = "l00754" > < / a > < span class = "lineno" > 754< / span > }< / div >
< / div >
2024-07-12 06:32:08 +08:00
< 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 = "}" >
2024-07-26 02:59:11 +08:00
< div class = "line" > < a id = "l00756" name = "l00756" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga7cc080a4f9d4a667f2099aa0dbfefadd" > 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 > (< 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 = "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, < a class = "code hl_class" href = "classmlx_1_1core_1_1array.html" > array< / a > (b));< / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00758" name = "l00758" > < / a > < span class = "lineno" > 758< / span > }< / div >
< / div >
2024-07-26 02:59:11 +08:00
< div class = "line" > < a id = "l00759" name = "l00759" > < / a > < span class = "lineno" > 759< / span > < / div >
< div class = "line" > < a id = "l00761" name = "l00761" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga196c240d3d0fcbb4713802c485e15133" > 761< / 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 = "l00762" name = "l00762" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga0c7f3cb36d4ca516c7a33142f88b9181" > 762< / 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 = "l00763" name = "l00763" > < / a > < span class = "lineno" > 763< / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "foldopen" id = "foldopen00764" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00764" name = "l00764" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gae68d3d0691ba951501218e98439f3465" > 764< / 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 = "l00765" name = "l00765" > < / a > < span class = "lineno" > 765< / 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 = "l00766" name = "l00766" > < / a > < span class = "lineno" > 766< / span > }< / div >
2024-07-12 06:32:08 +08:00
< / 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 = "}" >
2024-07-26 02:59:11 +08:00
< div class = "line" > < a id = "l00768" name = "l00768" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaf5e5d882c51ad0a0ea315c274d5439b2" > 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 > (< 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 = "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, < a class = "code hl_class" href = "classmlx_1_1core_1_1array.html" > array< / a > (b));< / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l00770" name = "l00770" > < / a > < span class = "lineno" > 770< / span > }< / div >
< / div >
2024-07-26 02:59:11 +08:00
< div class = "line" > < a id = "l00771" name = "l00771" > < / a > < span class = "lineno" > 771< / span > < / div >
< div class = "line" > < a id = "l00773" name = "l00773" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaf57392e641640b5d06e4c99518391c38" > 773< / 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 = "l00774" name = "l00774" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga26c33f5cdb6fc10d272acd6e208034e0" > 774< / 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 = "l00775" name = "l00775" > < / a > < span class = "lineno" > 775< / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "foldopen" id = "foldopen00776" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00776" name = "l00776" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gac22a67f7de797b1ae59029843cbdcab6" > 776< / 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 = "l00777" name = "l00777" > < / a > < span class = "lineno" > 777< / 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 = "l00778" name = "l00778" > < / a > < span class = "lineno" > 778< / span > }< / div >
< / 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#ga6f2369ed5fae8ff9b1528670a004dde2" > 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 > (< 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 = "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, < a class = "code hl_class" href = "classmlx_1_1core_1_1array.html" > array< / 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 > < / div >
< div class = "line" > < a id = "l00785" name = "l00785" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga77472dd06cfa7a30a42e4fd927bd859f" > 785< / 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 = "l00786" name = "l00786" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaeedf77f722b394429f1a7f6c367883bf" > 786< / 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 = "l00787" name = "l00787" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga7366ec7f453be2a4dc449f0faa1bf554" > 787< / 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 = "l00788" name = "l00788" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gadfb324ae9b4feb2c7ea0ac6ade639f38" > 788< / 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 = "l00789" name = "l00789" > < / a > < span class = "lineno" > 789< / span > < / div >
< div class = "line" > < a id = "l00791" name = "l00791" > < / a > < span class = "lineno" > 791< / span > std::vector< array> < / div >
< div class = "line" > < a id = "l00792" name = "l00792" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaa30ebc0a8376dbc3f7e46a47052b5894" > 792< / 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 = "l00793" name = "l00793" > < / a > < span class = "lineno" > 793< / span > < / div >
< div class = "line" > < a id = "l00795" name = "l00795" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga05b4c6054d028107869511f927da01cd" > 795< / 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 = "l00796" name = "l00796" > < / a > < span class = "lineno" > 796< / span > < / div >
< div class = "line" > < a id = "l00798" name = "l00798" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga99f5c904f724156a814d7817188351d2" > 798< / 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 = "l00799" name = "l00799" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gab3bfbf82b1e4de7b00bbcf1a2255fbde" > 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#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 = "l00800" name = "l00800" > < / a > < span class = "lineno" > 800< / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "foldopen" id = "foldopen00801" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00801" name = "l00801" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga50817666f0b82afcbf4a123486af9908" > 801< / 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 = "l00802" name = "l00802" > < / a > < span class = "lineno" > 802< / 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 = "l00803" name = "l00803" > < / a > < span class = "lineno" > 803< / span > }< / div >
< / 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#ga46c01daa07433542a477d216e13a8480" > 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 > (< 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 = "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, < a class = "code hl_class" href = "classmlx_1_1core_1_1array.html" > array< / 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 > < / div >
< div class = "line" > < a id = "l00810" name = "l00810" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga7ade2ea305e2e4219c3609443fb5db8d" > 810< / 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 = "l00811" name = "l00811" > < / a > < span class = "lineno" > 811< / span > < / div >
< div class = "line" > < a id = "l00813" name = "l00813" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga49ba00c090f81f331c91b0c97040bce0" > 813< / 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 = "l00814" name = "l00814" > < / a > < span class = "lineno" > 814< / span > < / div >
< div class = "line" > < a id = "l00816" name = "l00816" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga8d656904aa2690b60955ae745aecfc30" > 816< / 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 = "l00817" name = "l00817" > < / a > < span class = "lineno" > 817< / span > < / div >
< div class = "line" > < a id = "l00819" name = "l00819" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga1404ecceff83fd9b9139b7520f55e096" > 819< / 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 = "l00820" name = "l00820" > < / a > < span class = "lineno" > 820< / span > < / div >
< div class = "line" > < a id = "l00822" name = "l00822" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga1234e4c39cfa79f19d4bdb5b8ea4d45e" > 822< / 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 = "l00823" name = "l00823" > < / a > < span class = "lineno" > 823< / span > < / div >
< div class = "line" > < a id = "l00825" name = "l00825" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga8a3b04e23e347d99ecf411fd6f4e5125" > 825< / 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 = "l00826" name = "l00826" > < / a > < span class = "lineno" > 826< / span > < / div >
< div class = "line" > < a id = "l00828" name = "l00828" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaebf0a73ad3732fba39df37826c235692" > 828< / 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 = "l00829" name = "l00829" > < / a > < span class = "lineno" > 829< / span > < / div >
< div class = "line" > < a id = "l00831" name = "l00831" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga39dfdf72b556012aa35ff27a94116e74" > 831< / 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 = "l00832" name = "l00832" > < / a > < span class = "lineno" > 832< / span > < / div >
< div class = "line" > < a id = "l00834" name = "l00834" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga3f10e89a4bcb1a8fa44fb33b8d1176a5" > 834< / 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 = "l00835" name = "l00835" > < / a > < span class = "lineno" > 835< / span > < / div >
< div class = "line" > < a id = "l00837" name = "l00837" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga8770e8c8f23f13343911f4c9d6e1c619" > 837< / 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 = "l00838" name = "l00838" > < / a > < span class = "lineno" > 838< / span > < / div >
< div class = "line" > < a id = "l00840" name = "l00840" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga08bec7cb10c84466487b507fc5bf9776" > 840< / 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 = "l00841" name = "l00841" > < / a > < span class = "lineno" > 841< / span > < / div >
< div class = "line" > < a id = "l00843" name = "l00843" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaa041f3f070e68f4946db07516b7d092e" > 843< / 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 = "l00844" name = "l00844" > < / a > < span class = "lineno" > 844< / span > < / div >
< div class = "line" > < a id = "l00846" name = "l00846" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga6caba9c92b5989123501f909cc7da354" > 846< / 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 = "l00847" name = "l00847" > < / a > < span class = "lineno" > 847< / span > < / div >
< div class = "line" > < a id = "l00849" name = "l00849" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaf532375c6563dbd6e329bdedf0224dd7" > 849< / 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 = "l00850" name = "l00850" > < / a > < span class = "lineno" > 850< / span > < / div >
< div class = "line" > < a id = "l00852" name = "l00852" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga2181b71cda88007a3092be4795ff0715" > 852< / 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 = "l00853" name = "l00853" > < / a > < span class = "lineno" > 853< / span > < / div >
< div class = "line" > < a id = "l00855" name = "l00855" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga5efb19aa0dfa42d8a3d5e1dfd569cd6d" > 855< / 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 = "l00856" name = "l00856" > < / a > < span class = "lineno" > 856< / span > < / div >
< div class = "line" > < a id = "l00858" name = "l00858" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gac62e2cedc49ef2c90dd8584000317450" > 858< / 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 = "l00859" name = "l00859" > < / a > < span class = "lineno" > 859< / span > < / div >
< div class = "line" > < a id = "l00861" name = "l00861" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaafafcfcebdf7248679c8543d0c0497e5" > 861< / 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 = "l00862" name = "l00862" > < / a > < span class = "lineno" > 862< / span > < / div >
< div class = "line" > < a id = "l00864" name = "l00864" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gab46a35925a04c5a9d2ec7898ee55358e" > 864< / 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 = "l00865" name = "l00865" > < / a > < span class = "lineno" > 865< / span > < / div >
< div class = "line" > < a id = "l00867" name = "l00867" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga3a70569b50e1083c5ded199d73fb960c" > 867< / 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 = "l00868" name = "l00868" > < / a > < span class = "lineno" > 868< / span > < / div >
< div class = "line" > < a id = "l00870" name = "l00870" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga07aa8059adba5b9a8818027b8aafd31e" > 870< / 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 = "l00871" name = "l00871" > < / a > < span class = "lineno" > 871< / span > < / div >
< div class = "line" > < a id = "l00873" name = "l00873" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga6fb22d4926133573e430fcc92f4eef31" > 873< / 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 = "l00874" name = "l00874" > < / a > < span class = "lineno" > 874< / span > < / div >
< div class = "line" > < a id = "l00876" name = "l00876" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga144228d7222d15af3a135b8b0f3fa21b" > 876< / 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 = "l00877" name = "l00877" > < / a > < span class = "lineno" > 877< / span > < / div >
< div class = "line" > < a id = "l00879" name = "l00879" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga1fdcc7fc8819caf2e6f1c327ed4e9b9e" > 879< / 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 = "l00880" name = "l00880" > < / a > < span class = "lineno" > 880< / span > < / div >
< div class = "line" > < a id = "l00882" name = "l00882" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga20a1f4270c35b0fa544f5105a87a1604" > 882< / 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 = "l00883" name = "l00883" > < / a > < span class = "lineno" > 883< / span > < / div >
< div class = "line" > < a id = "l00885" name = "l00885" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaf985df6609c6bd75a14a844655d89eaa" > 885< / 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 = "l00886" name = "l00886" > < / a > < span class = "lineno" > 886< / span > < / div >
< div class = "line" > < a id = "l00888" name = "l00888" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga708abf8f79609cd6831db7c38cafac0e" > 888< / 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 = "l00889" name = "l00889" > < / a > < span class = "lineno" > 889< / span > < / div >
< div class = "line" > < a id = "l00891" name = "l00891" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga292a335240fd5d6d625fb7a340ff5eb0" > 891< / 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 = "l00892" name = "l00892" > < / a > < span class = "lineno" > 892< / span > < / div >
< div class = "line" > < a id = "l00894" name = "l00894" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga76fb9062c64264e34d2e07013390557c" > 894< / 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 = "l00895" name = "l00895" > < / a > < span class = "lineno" > 895< / span > < / div >
< div class = "line" > < a id = "l00897" name = "l00897" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga54ca54f06bfb2be15b163a5209e2a0f0" > 897< / 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 = "l00898" name = "l00898" > < / a > < span class = "lineno" > 898< / span > < / div >
< div class = "line" > < a id = "l00900" name = "l00900" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga36bc28f1deb2fe668ca9ae1e447b6b1f" > 900< / 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 = "l00901" name = "l00901" > < / a > < span class = "lineno" > 901< / span > < / div >
< div class = "line" > < a id = "l00903" name = "l00903" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga2d74d43f007a069384e89d8416525331" > 903< / 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 = "foldopen00904" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00904" name = "l00904" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaf18fb7e98bf8cf3b7fbc5e64c988a95b" > 904< / 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 = "l00905" name = "l00905" > < / a > < span class = "lineno" > 905< / 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 = "l00906" name = "l00906" > < / a > < span class = "lineno" > 906< / span > }< / div >
< / div >
< div class = "line" > < a id = "l00907" name = "l00907" > < / a > < span class = "lineno" > 907< / span > < / div >
< div class = "line" > < a id = "l00909" name = "l00909" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga753d59f5a9f5f2362865ee83b4dced2a" > 909< / 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 = "l00910" name = "l00910" > < / a > < span class = "lineno" > 910< / span > < / div >
< div class = "line" > < a id = "l00912" name = "l00912" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gab6e7f655a9ff15350ca5379692f9d444" > 912< / 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 = "l00913" name = "l00913" > < / a > < span class = "lineno" > 913< / 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 = "l00914" name = "l00914" > < / a > < span class = "lineno" > 914< / span > < span class = "keyword" > const< / span > std::vector< array> & indices,< / div >
< div class = "line" > < a id = "l00915" name = "l00915" > < / a > < span class = "lineno" > 915< / span > < span class = "keyword" > const< / span > std::vector< int> & axes,< / div >
< div class = "line" > < a id = "l00916" name = "l00916" > < / a > < span class = "lineno" > 916< / span > < span class = "keyword" > const< / span > std::vector< int> & slice_sizes,< / div >
< div class = "line" > < a id = "l00917" name = "l00917" > < / a > < span class = "lineno" > 917< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "foldopen" id = "foldopen00918" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l00918" name = "l00918" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gadb4337ca5d4f88fe9e7c083bc478158b" > 918< / 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 = "l00919" name = "l00919" > < / a > < span class = "lineno" > 919< / 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 = "l00920" name = "l00920" > < / a > < span class = "lineno" > 920< / 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 = "l00921" name = "l00921" > < / a > < span class = "lineno" > 921< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l00922" name = "l00922" > < / a > < span class = "lineno" > 922< / span > < span class = "keyword" > const< / span > std::vector< int> & slice_sizes,< / div >
< div class = "line" > < a id = "l00923" name = "l00923" > < / a > < span class = "lineno" > 923< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {}) {< / div >
< div class = "line" > < a id = "l00924" name = "l00924" > < / a > < span class = "lineno" > 924< / 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 = "l00925" name = "l00925" > < / a > < span class = "lineno" > 925< / span > }< / div >
< / div >
< div class = "line" > < a id = "l00926" name = "l00926" > < / a > < span class = "lineno" > 926< / span > < / div >
< div class = "line" > < a id = "l00928" name = "l00928" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gac2fc270882fcfa81eb8bd068cc0d86d7" > 928< / 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 = "l00929" name = "l00929" > < / a > < span class = "lineno" > 929< / 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 = "l00930" name = "l00930" > < / a > < span class = "lineno" > 930< / 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 = "l00931" name = "l00931" > < / a > < span class = "lineno" > 931< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l00932" name = "l00932" > < / a > < span class = "lineno" > 932< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00933" name = "l00933" > < / a > < span class = "lineno" > 933< / span > < / div >
< div class = "line" > < a id = "l00935" name = "l00935" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga45d0f423a5e030440ef753f36c5aabf1" > 935< / 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 = "l00936" name = "l00936" > < / a > < span class = "lineno" > 936< / span > < / div >
< div class = "line" > < a id = "l00938" name = "l00938" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gae0a81d4f983e296a87302e36d65bfc76" > 938< / 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 = "l00939" name = "l00939" > < / a > < span class = "lineno" > 939< / 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 = "l00940" name = "l00940" > < / a > < span class = "lineno" > 940< / 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 = "l00941" name = "l00941" > < / a > < span class = "lineno" > 941< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l00942" name = "l00942" > < / a > < span class = "lineno" > 942< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l00943" name = "l00943" > < / a > < span class = "lineno" > 943< / span > < / div >
< div class = "line" > < a id = "l01042" name = "l01042" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gad438be8f90bae9d37c6853b8f4225d61" > 1042< / 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 = "l01043" name = "l01043" > < / a > < span class = "lineno" > 1043< / 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 = "l01044" name = "l01044" > < / a > < span class = "lineno" > 1044< / span > < span class = "keyword" > const< / span > std::vector< array> & indices,< / div >
< div class = "line" > < a id = "l01045" name = "l01045" > < / a > < span class = "lineno" > 1045< / 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 = "l01046" name = "l01046" > < / a > < span class = "lineno" > 1046< / span > < span class = "keyword" > const< / span > std::vector< int> & axes,< / div >
< div class = "line" > < a id = "l01047" name = "l01047" > < / a > < span class = "lineno" > 1047< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "foldopen" id = "foldopen01048" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l01048" name = "l01048" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gac2c2b379a3ce959dbe1c4a68f112edfe" > 1048< / 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 = "l01049" name = "l01049" > < / a > < span class = "lineno" > 1049< / 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 = "l01050" name = "l01050" > < / a > < span class = "lineno" > 1050< / 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 = "l01051" name = "l01051" > < / a > < span class = "lineno" > 1051< / 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 = "l01052" name = "l01052" > < / a > < span class = "lineno" > 1052< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l01053" name = "l01053" > < / a > < span class = "lineno" > 1053< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {}) {< / div >
< div class = "line" > < a id = "l01054" name = "l01054" > < / a > < span class = "lineno" > 1054< / 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 = "l01055" name = "l01055" > < / a > < span class = "lineno" > 1055< / span > }< / div >
< / div >
< div class = "line" > < a id = "l01056" name = "l01056" > < / a > < span class = "lineno" > 1056< / span > < / div >
< div class = "line" > < a id = "l01058" name = "l01058" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gacd14c2b5cfebf343fc2d672722f8d174" > 1058< / 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 = "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 > & a,< / div >
< div class = "line" > < a id = "l01060" name = "l01060" > < / a > < span class = "lineno" > 1060< / span > < span class = "keyword" > const< / span > std::vector< array> & indices,< / div >
< div class = "line" > < a id = "l01061" name = "l01061" > < / a > < span class = "lineno" > 1061< / 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 = "l01062" name = "l01062" > < / a > < span class = "lineno" > 1062< / span > < span class = "keyword" > const< / span > std::vector< int> & axes,< / div >
< div class = "line" > < a id = "l01063" name = "l01063" > < / a > < span class = "lineno" > 1063< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "foldopen" id = "foldopen01064" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l01064" name = "l01064" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gac13318518e5703f1273c5366eb523a5a" > 1064< / 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 = "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 > & a,< / div >
< div class = "line" > < a id = "l01066" name = "l01066" > < / a > < span class = "lineno" > 1066< / 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 = "l01067" name = "l01067" > < / a > < span class = "lineno" > 1067< / 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 = "l01068" name = "l01068" > < / a > < span class = "lineno" > 1068< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l01069" name = "l01069" > < / a > < span class = "lineno" > 1069< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {}) {< / div >
< div class = "line" > < a id = "l01070" name = "l01070" > < / a > < span class = "lineno" > 1070< / 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 = "l01071" name = "l01071" > < / a > < span class = "lineno" > 1071< / span > }< / div >
< / div >
< div class = "line" > < a id = "l01072" name = "l01072" > < / a > < span class = "lineno" > 1072< / span > < / div >
< div class = "line" > < a id = "l01074" name = "l01074" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga3708b5bcb61e2c63d213c4ce6ad0ffc0" > 1074< / 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 = "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 > & a,< / div >
< div class = "line" > < a id = "l01076" name = "l01076" > < / a > < span class = "lineno" > 1076< / span > < span class = "keyword" > const< / span > std::vector< array> & indices,< / div >
< div class = "line" > < a id = "l01077" name = "l01077" > < / a > < span class = "lineno" > 1077< / 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 = "l01078" name = "l01078" > < / a > < span class = "lineno" > 1078< / span > < span class = "keyword" > const< / span > std::vector< int> & axes,< / div >
< div class = "line" > < a id = "l01079" name = "l01079" > < / a > < span class = "lineno" > 1079< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "foldopen" id = "foldopen01080" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l01080" name = "l01080" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaf83c53c453faa9083ba27e4b97539339" > 1080< / 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 = "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 > & a,< / div >
< div class = "line" > < a id = "l01082" name = "l01082" > < / a > < span class = "lineno" > 1082< / 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 = "l01083" name = "l01083" > < / a > < span class = "lineno" > 1083< / 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 = "l01084" name = "l01084" > < / a > < span class = "lineno" > 1084< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l01085" name = "l01085" > < / a > < span class = "lineno" > 1085< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {}) {< / div >
< div class = "line" > < a id = "l01086" name = "l01086" > < / a > < span class = "lineno" > 1086< / 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 = "l01087" name = "l01087" > < / a > < span class = "lineno" > 1087< / span > }< / div >
< / div >
< div class = "line" > < a id = "l01088" name = "l01088" > < / a > < span class = "lineno" > 1088< / span > < / div >
< div class = "line" > < a id = "l01090" name = "l01090" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga05881a4157cd113c9392d168a79e6673" > 1090< / 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 = "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 > & a,< / div >
< div class = "line" > < a id = "l01092" name = "l01092" > < / a > < span class = "lineno" > 1092< / span > < span class = "keyword" > const< / span > std::vector< array> & indices,< / div >
< div class = "line" > < a id = "l01093" name = "l01093" > < / a > < span class = "lineno" > 1093< / 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 = "l01094" name = "l01094" > < / a > < span class = "lineno" > 1094< / span > < span class = "keyword" > const< / span > std::vector< int> & axes,< / div >
< div class = "line" > < a id = "l01095" name = "l01095" > < / a > < span class = "lineno" > 1095< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "foldopen" id = "foldopen01096" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l01096" name = "l01096" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga9adda5f9202bb3486e4d9e1114e3a56f" > 1096< / 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 = "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 > & a,< / div >
< div class = "line" > < a id = "l01098" name = "l01098" > < / a > < span class = "lineno" > 1098< / 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 = "l01099" name = "l01099" > < / a > < span class = "lineno" > 1099< / 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 = "l01100" name = "l01100" > < / a > < span class = "lineno" > 1100< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l01101" name = "l01101" > < / a > < span class = "lineno" > 1101< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {}) {< / div >
< div class = "line" > < a id = "l01102" name = "l01102" > < / a > < span class = "lineno" > 1102< / 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 = "l01103" name = "l01103" > < / a > < span class = "lineno" > 1103< / span > }< / div >
< / div >
< div class = "line" > < a id = "l01105" name = "l01105" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga0ca16b7579dfc899f3f7fd40245ba7c5" > 1105< / 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 = "l01106" name = "l01106" > < / a > < span class = "lineno" > 1106< / 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 = "l01107" name = "l01107" > < / a > < span class = "lineno" > 1107< / span > < span class = "keyword" > const< / span > std::vector< array> & indices,< / div >
< div class = "line" > < a id = "l01108" name = "l01108" > < / a > < span class = "lineno" > 1108< / 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 = "l01109" name = "l01109" > < / a > < span class = "lineno" > 1109< / span > < span class = "keyword" > const< / span > std::vector< int> & axes,< / div >
< div class = "line" > < a id = "l01110" name = "l01110" > < / a > < span class = "lineno" > 1110< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "foldopen" id = "foldopen01111" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l01111" name = "l01111" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga51fa762a997c243ca7a19e1ed3e83199" > 1111< / 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 = "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 > & a,< / 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 > & indices,< / div >
< div class = "line" > < a id = "l01114" name = "l01114" > < / a > < span class = "lineno" > 1114< / 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 = "l01115" name = "l01115" > < / a > < span class = "lineno" > 1115< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l01116" name = "l01116" > < / a > < span class = "lineno" > 1116< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {}) {< / div >
< div class = "line" > < a id = "l01117" name = "l01117" > < / a > < span class = "lineno" > 1117< / 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 = "l01118" name = "l01118" > < / a > < span class = "lineno" > 1118< / span > }< / div >
< / div >
< div class = "line" > < a id = "l01119" name = "l01119" > < / a > < span class = "lineno" > 1119< / span > < / div >
< div class = "line" > < a id = "l01121" name = "l01121" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga297f853b3d90ec8ae81263977ba2ddb1" > 1121< / 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 = "l01122" name = "l01122" > < / a > < span class = "lineno" > 1122< / span > < / div >
< div class = "line" > < a id = "l01124" name = "l01124" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga102f23aa0b0c3d3296a321c694617aa1" > 1124< / 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 = "l01125" name = "l01125" > < / a > < span class = "lineno" > 1125< / span > < / div >
< div class = "line" > < a id = "l01127" name = "l01127" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6" > 1127< / 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 = "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 > & a,< / div >
< div class = "line" > < a id = "l01129" name = "l01129" > < / a > < span class = "lineno" > 1129< / span > < span class = "keyword" > const< / span > std::vector< int> & axes,< / div >
< div class = "line" > < a id = "l01130" name = "l01130" > < / a > < span class = "lineno" > 1130< / span > < span class = "keywordtype" > bool< / span > precise = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l01131" name = "l01131" > < / a > < span class = "lineno" > 1131< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01132" name = "l01132" > < / a > < span class = "lineno" > 1132< / span > < / div >
< div class = "line" > < a id = "l01134" name = "l01134" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga1ae3614d07d873892a530d14c3857d0b" > 1134< / 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 = "l01135" name = "l01135" > < / a > < span class = "lineno" > 1135< / span > < / div >
< div class = "line" > < a id = "l01137" name = "l01137" > < / a > < span class = "lineno" > 1137< / span > < span class = "keyword" > inline< / span > array< / div >
< div class = "foldopen" id = "foldopen01138" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l01138" name = "l01138" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga06f570d73716a24303e6de3aaba4457b" > 1138< / 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 = "l01139" name = "l01139" > < / a > < span class = "lineno" > 1139< / 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 = "l01140" name = "l01140" > < / a > < span class = "lineno" > 1140< / span > }< / div >
< / div >
< div class = "line" > < a id = "l01141" name = "l01141" > < / a > < span class = "lineno" > 1141< / span > < / div >
< div class = "line" > < a id = "l01143" name = "l01143" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga7972058715c26559dff9c9ae2a3ef76d" > 1143< / 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 >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l01144" name = "l01144" > < / a > < span class = "lineno" > 1144< / span > < / div >
2024-07-26 02:59:11 +08:00
< div class = "line" > < a id = "l01146" name = "l01146" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaddc825a5c173e195ab0fda83ad630420" > 1146< / 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 >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l01147" name = "l01147" > < / a > < span class = "lineno" > 1147< / 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 = "l01148" name = "l01148" > < / a > < span class = "lineno" > 1148< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l01149" name = "l01149" > < / a > < span class = "lineno" > 1149< / span > < span class = "keywordtype" > bool< / span > reverse = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l01150" name = "l01150" > < / a > < span class = "lineno" > 1150< / span > < span class = "keywordtype" > bool< / span > inclusive = < span class = "keyword" > true< / span > ,< / div >
< div class = "line" > < a id = "l01151" name = "l01151" > < / a > < span class = "lineno" > 1151< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01152" name = "l01152" > < / a > < span class = "lineno" > 1152< / span > < / div >
2024-07-26 02:59:11 +08:00
< div class = "line" > < a id = "l01154" name = "l01154" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga0d71dfbc14ef3ed564b0c5ee26af680f" > 1154< / 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 >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l01155" name = "l01155" > < / a > < span class = "lineno" > 1155< / 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 = "l01156" name = "l01156" > < / a > < span class = "lineno" > 1156< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l01157" name = "l01157" > < / a > < span class = "lineno" > 1157< / span > < span class = "keywordtype" > bool< / span > reverse = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l01158" name = "l01158" > < / a > < span class = "lineno" > 1158< / span > < span class = "keywordtype" > bool< / span > inclusive = < span class = "keyword" > true< / span > ,< / div >
< div class = "line" > < a id = "l01159" name = "l01159" > < / a > < span class = "lineno" > 1159< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01160" name = "l01160" > < / a > < span class = "lineno" > 1160< / span > < / div >
2024-07-26 02:59:11 +08:00
< div class = "line" > < a id = "l01162" name = "l01162" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaee37cac8476e8f8d666bcded5bc59143" > 1162< / 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 >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l01163" name = "l01163" > < / a > < span class = "lineno" > 1163< / 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 = "l01164" name = "l01164" > < / a > < span class = "lineno" > 1164< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l01165" name = "l01165" > < / a > < span class = "lineno" > 1165< / span > < span class = "keywordtype" > bool< / span > reverse = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l01166" name = "l01166" > < / a > < span class = "lineno" > 1166< / span > < span class = "keywordtype" > bool< / span > inclusive = < span class = "keyword" > true< / span > ,< / div >
< div class = "line" > < a id = "l01167" name = "l01167" > < / a > < span class = "lineno" > 1167< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01168" name = "l01168" > < / a > < span class = "lineno" > 1168< / span > < / div >
2024-07-26 02:59:11 +08:00
< div class = "line" > < a id = "l01170" name = "l01170" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga19c1bf6929fe8d66b9cd408946aea6a8" > 1170< / 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 = "l01171" name = "l01171" > < / a > < span class = "lineno" > 1171< / 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 = "l01172" name = "l01172" > < / a > < span class = "lineno" > 1172< / span > < span class = "keywordtype" > int< / span > axis,< / div >
< div class = "line" > < a id = "l01173" name = "l01173" > < / a > < span class = "lineno" > 1173< / span > < span class = "keywordtype" > bool< / span > reverse = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l01174" name = "l01174" > < / a > < span class = "lineno" > 1174< / span > < span class = "keywordtype" > bool< / span > inclusive = < span class = "keyword" > true< / span > ,< / div >
< div class = "line" > < a id = "l01175" name = "l01175" > < / a > < span class = "lineno" > 1175< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01176" name = "l01176" > < / a > < span class = "lineno" > 1176< / span > < / div >
< div class = "line" > < a id = "l01178" name = "l01178" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e" > 1178< / 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 = "l01179" name = "l01179" > < / a > < span class = "lineno" > 1179< / span > < a class = "code hl_class" href = "classmlx_1_1core_1_1array.html" > array< / a > input,< / div >
< div class = "line" > < a id = "l01180" name = "l01180" > < / a > < span class = "lineno" > 1180< / span > < a class = "code hl_class" href = "classmlx_1_1core_1_1array.html" > array< / a > weight,< / div >
< div class = "line" > < a id = "l01181" name = "l01181" > < / a > < span class = "lineno" > 1181< / span > std::vector< int> stride = {},< / div >
< div class = "line" > < a id = "l01182" name = "l01182" > < / a > < span class = "lineno" > 1182< / span > std::vector< int> padding_lo = {},< / div >
< div class = "line" > < a id = "l01183" name = "l01183" > < / a > < span class = "lineno" > 1183< / span > std::vector< int> padding_hi = {},< / div >
< div class = "line" > < a id = "l01184" name = "l01184" > < / a > < span class = "lineno" > 1184< / span > std::vector< int> kernel_dilation = {},< / div >
< div class = "line" > < a id = "l01185" name = "l01185" > < / a > < span class = "lineno" > 1185< / span > std::vector< int> input_dilation = {},< / div >
< div class = "line" > < a id = "l01186" name = "l01186" > < / a > < span class = "lineno" > 1186< / span > < span class = "keywordtype" > int< / span > groups = 1,< / div >
< div class = "line" > < a id = "l01187" name = "l01187" > < / a > < span class = "lineno" > 1187< / span > < span class = "keywordtype" > bool< / span > flip = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l01188" name = "l01188" > < / a > < span class = "lineno" > 1188< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01189" name = "l01189" > < / a > < span class = "lineno" > 1189< / span > < / div >
< div class = "foldopen" id = "foldopen01191" data-start = "{" data-end = "}" >
< div class = "line" > < a id = "l01191" name = "l01191" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gab59f89942cd1efaadffe9e8762e3c99d" > 1191< / 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 = "l01192" name = "l01192" > < / a > < span class = "lineno" > 1192< / 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 = "l01193" name = "l01193" > < / a > < span class = "lineno" > 1193< / 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 = "l01194" name = "l01194" > < / a > < span class = "lineno" > 1194< / span > std::vector< int> stride = {},< / div >
< div class = "line" > < a id = "l01195" name = "l01195" > < / a > < span class = "lineno" > 1195< / span > std::vector< int> padding = {},< / div >
< div class = "line" > < a id = "l01196" name = "l01196" > < / a > < span class = "lineno" > 1196< / span > std::vector< int> kernel_dilation = {},< / div >
< div class = "line" > < a id = "l01197" name = "l01197" > < / a > < span class = "lineno" > 1197< / span > std::vector< int> input_dilation = {},< / div >
< div class = "line" > < a id = "l01198" name = "l01198" > < / a > < span class = "lineno" > 1198< / span > < span class = "keywordtype" > int< / span > groups = 1,< / div >
< div class = "line" > < a id = "l01199" name = "l01199" > < / a > < span class = "lineno" > 1199< / span > < span class = "keywordtype" > bool< / span > flip = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a id = "l01200" name = "l01200" > < / a > < span class = "lineno" > 1200< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {}) {< / div >
< div class = "line" > < a id = "l01201" name = "l01201" > < / a > < span class = "lineno" > 1201< / 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 = "l01202" name = "l01202" > < / a > < span class = "lineno" > 1202< / span > < span class = "comment" > /* const array& input = */< / span > input,< / div >
< div class = "line" > < a id = "l01203" name = "l01203" > < / a > < span class = "lineno" > 1203< / span > < span class = "comment" > /* const array& weight = */< / span > weight,< / div >
< div class = "line" > < a id = "l01204" name = "l01204" > < / a > < span class = "lineno" > 1204< / span > < span class = "comment" > /* std::vector< int> stride = */< / span > stride,< / div >
< div class = "line" > < a id = "l01205" name = "l01205" > < / a > < span class = "lineno" > 1205< / span > < span class = "comment" > /* std::vector< int> padding_lo = */< / span > padding,< / div >
< div class = "line" > < a id = "l01206" name = "l01206" > < / a > < span class = "lineno" > 1206< / span > < span class = "comment" > /* std::vector< int> padding_hi = */< / span > padding,< / div >
< div class = "line" > < a id = "l01207" name = "l01207" > < / a > < span class = "lineno" > 1207< / span > < span class = "comment" > /* std::vector< int> kernel_dilation = */< / span > kernel_dilation,< / div >
< div class = "line" > < a id = "l01208" name = "l01208" > < / a > < span class = "lineno" > 1208< / span > < span class = "comment" > /* std::vector< int> input_dilation = */< / span > input_dilation,< / div >
< div class = "line" > < a id = "l01209" name = "l01209" > < / a > < span class = "lineno" > 1209< / span > < span class = "comment" > /* int groups = */< / span > groups,< / div >
< div class = "line" > < a id = "l01210" name = "l01210" > < / a > < span class = "lineno" > 1210< / span > < span class = "comment" > /* bool flip = */< / span > flip,< / div >
< div class = "line" > < a id = "l01211" name = "l01211" > < / a > < span class = "lineno" > 1211< / span > < span class = "comment" > /* StreamOrDevice s = */< / span > s);< / div >
< div class = "line" > < a id = "l01212" name = "l01212" > < / a > < span class = "lineno" > 1212< / span > }< / div >
< / div >
< div class = "line" > < a id = "l01213" name = "l01213" > < / a > < span class = "lineno" > 1213< / span > < / div >
< div class = "line" > < a id = "l01215" name = "l01215" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga30d47e08093c03a3676f235f9f559411" > 1215< / 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 = "l01216" name = "l01216" > < / a > < span class = "lineno" > 1216< / 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 = "l01217" name = "l01217" > < / a > < span class = "lineno" > 1217< / 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 = "l01218" name = "l01218" > < / a > < span class = "lineno" > 1218< / span > < span class = "keywordtype" > int< / span > stride = 1,< / div >
< div class = "line" > < a id = "l01219" name = "l01219" > < / a > < span class = "lineno" > 1219< / span > < span class = "keywordtype" > int< / span > padding = 0,< / div >
< div class = "line" > < a id = "l01220" name = "l01220" > < / a > < span class = "lineno" > 1220< / span > < span class = "keywordtype" > int< / span > dilation = 1,< / div >
< div class = "line" > < a id = "l01221" name = "l01221" > < / a > < span class = "lineno" > 1221< / span > < span class = "keywordtype" > int< / span > groups = 1,< / div >
< div class = "line" > < a id = "l01222" name = "l01222" > < / a > < span class = "lineno" > 1222< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01223" name = "l01223" > < / a > < span class = "lineno" > 1223< / span > < / div >
< div class = "line" > < a id = "l01225" name = "l01225" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga73b02833229678786e7f302d458d5a83" > 1225< / 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 = "l01226" name = "l01226" > < / a > < span class = "lineno" > 1226< / 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 = "l01227" name = "l01227" > < / a > < span class = "lineno" > 1227< / 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 = "l01228" name = "l01228" > < / a > < span class = "lineno" > 1228< / span > < span class = "keyword" > const< / span > std::pair< int, int> & stride = {1, 1},< / div >
< div class = "line" > < a id = "l01229" name = "l01229" > < / a > < span class = "lineno" > 1229< / span > < span class = "keyword" > const< / span > std::pair< int, int> & padding = {0, 0},< / div >
< div class = "line" > < a id = "l01230" name = "l01230" > < / a > < span class = "lineno" > 1230< / span > < span class = "keyword" > const< / span > std::pair< int, int> & dilation = {1, 1},< / div >
< div class = "line" > < a id = "l01231" name = "l01231" > < / a > < span class = "lineno" > 1231< / span > < span class = "keywordtype" > int< / span > groups = 1,< / div >
< div class = "line" > < a id = "l01232" name = "l01232" > < / a > < span class = "lineno" > 1232< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01233" name = "l01233" > < / a > < span class = "lineno" > 1233< / span > < / div >
< div class = "line" > < a id = "l01235" name = "l01235" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga6e9907d2f14dc4803e4306b3dbc4b3ca" > 1235< / 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 = "l01236" name = "l01236" > < / a > < span class = "lineno" > 1236< / 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 = "l01237" name = "l01237" > < / a > < span class = "lineno" > 1237< / 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 = "l01238" name = "l01238" > < / a > < span class = "lineno" > 1238< / span > < span class = "keyword" > const< / span > std::tuple< int, int, int> & stride = {1, 1, 1},< / div >
< div class = "line" > < a id = "l01239" name = "l01239" > < / a > < span class = "lineno" > 1239< / span > < span class = "keyword" > const< / span > std::tuple< int, int, int> & padding = {0, 0, 0},< / div >
< div class = "line" > < a id = "l01240" name = "l01240" > < / a > < span class = "lineno" > 1240< / span > < span class = "keyword" > const< / span > std::tuple< int, int, int> & dilation = {1, 1, 1},< / div >
< div class = "line" > < a id = "l01241" name = "l01241" > < / a > < span class = "lineno" > 1241< / span > < span class = "keywordtype" > int< / span > groups = 1,< / div >
< div class = "line" > < a id = "l01242" name = "l01242" > < / a > < span class = "lineno" > 1242< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01243" name = "l01243" > < / a > < span class = "lineno" > 1243< / span > < / div >
< div class = "line" > < a id = "l01245" name = "l01245" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga6381c7062b232ac21f3fdc3136aaf86e" > 1245< / 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 = "l01246" name = "l01246" > < / a > < span class = "lineno" > 1246< / 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 = "l01247" name = "l01247" > < / a > < span class = "lineno" > 1247< / 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 = "l01248" name = "l01248" > < / a > < span class = "lineno" > 1248< / 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 = "l01249" name = "l01249" > < / a > < span class = "lineno" > 1249< / 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 = "l01250" name = "l01250" > < / a > < span class = "lineno" > 1250< / 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 = "l01251" name = "l01251" > < / a > < span class = "lineno" > 1251< / span > < span class = "keywordtype" > int< / span > group_size = 64,< / div >
< div class = "line" > < a id = "l01252" name = "l01252" > < / a > < span class = "lineno" > 1252< / span > < span class = "keywordtype" > int< / span > bits = 4,< / div >
< div class = "line" > < a id = "l01253" name = "l01253" > < / a > < span class = "lineno" > 1253< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01254" name = "l01254" > < / a > < span class = "lineno" > 1254< / span > < / div >
< div class = "line" > < a id = "l01256" name = "l01256" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gab43cc28690da7cdd43b43065adbd31da" > 1256< / 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 = "l01257" name = "l01257" > < / a > < span class = "lineno" > 1257< / 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 = "l01258" name = "l01258" > < / a > < span class = "lineno" > 1258< / span > < span class = "keywordtype" > int< / span > group_size = 64,< / div >
< div class = "line" > < a id = "l01259" name = "l01259" > < / a > < span class = "lineno" > 1259< / span > < span class = "keywordtype" > int< / span > bits = 4,< / div >
< div class = "line" > < a id = "l01260" name = "l01260" > < / a > < span class = "lineno" > 1260< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01261" name = "l01261" > < / a > < span class = "lineno" > 1261< / span > < / div >
< div class = "line" > < a id = "l01263" name = "l01263" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gabff758a5c1ce32ad7e8b78aba0164077" > 1263< / 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 = "l01264" name = "l01264" > < / a > < span class = "lineno" > 1264< / 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 = "l01265" name = "l01265" > < / a > < span class = "lineno" > 1265< / 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 = "l01266" name = "l01266" > < / a > < span class = "lineno" > 1266< / 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 = "l01267" name = "l01267" > < / a > < span class = "lineno" > 1267< / span > < span class = "keywordtype" > int< / span > group_size = 64,< / div >
< div class = "line" > < a id = "l01268" name = "l01268" > < / a > < span class = "lineno" > 1268< / span > < span class = "keywordtype" > int< / span > bits = 4,< / div >
< div class = "line" > < a id = "l01269" name = "l01269" > < / a > < span class = "lineno" > 1269< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01270" name = "l01270" > < / a > < span class = "lineno" > 1270< / span > < / div >
< div class = "line" > < a id = "l01272" name = "l01272" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga368a0dc0e5dfb76922e7aa55a95f12f0" > 1272< / 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 = "l01273" name = "l01273" > < / a > < span class = "lineno" > 1273< / 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 = "l01274" name = "l01274" > < / a > < span class = "lineno" > 1274< / 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 = "l01275" name = "l01275" > < / a > < span class = "lineno" > 1275< / 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 = "l01276" name = "l01276" > < / a > < span class = "lineno" > 1276< / 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 = "l01277" name = "l01277" > < / a > < span class = "lineno" > 1277< / span > std::optional< array> lhs_indices = std::nullopt,< / div >
< div class = "line" > < a id = "l01278" name = "l01278" > < / a > < span class = "lineno" > 1278< / span > std::optional< array> rhs_indices = std::nullopt,< / div >
< div class = "line" > < a id = "l01279" name = "l01279" > < / a > < span class = "lineno" > 1279< / 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 = "l01280" name = "l01280" > < / a > < span class = "lineno" > 1280< / span > < span class = "keywordtype" > int< / span > group_size = 64,< / div >
< div class = "line" > < a id = "l01281" name = "l01281" > < / a > < span class = "lineno" > 1281< / span > < span class = "keywordtype" > int< / span > bits = 4,< / div >
< div class = "line" > < a id = "l01282" name = "l01282" > < / a > < span class = "lineno" > 1282< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01283" name = "l01283" > < / a > < span class = "lineno" > 1283< / span > < / div >
< div class = "line" > < a id = "l01285" name = "l01285" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaf5c9735f4690327e1500e04e728fae70" > 1285< / 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 = "l01286" name = "l01286" > < / a > < span class = "lineno" > 1286< / 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 = "l01287" name = "l01287" > < / a > < span class = "lineno" > 1287< / 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 = "l01288" name = "l01288" > < / a > < span class = "lineno" > 1288< / span > < span class = "keyword" > const< / span > < span class = "keywordtype" > int< / span > axis = 2,< / div >
< div class = "line" > < a id = "l01289" name = "l01289" > < / a > < span class = "lineno" > 1289< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01290" name = "l01290" > < / a > < span class = "lineno" > 1290< / span > < / div >
< div class = "line" > < a id = "l01291" name = "l01291" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gad7fe00b566f89d607639c1a497cabbc6" > 1291< / 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 = "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 > & a,< / 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 > & b,< / div >
< div class = "line" > < a id = "l01294" name = "l01294" > < / a > < span class = "lineno" > 1294< / span > < span class = "keyword" > const< / span > std::vector< int> & axes_a,< / div >
< div class = "line" > < a id = "l01295" name = "l01295" > < / a > < span class = "lineno" > 1295< / span > < span class = "keyword" > const< / span > std::vector< int> & axes_b,< / div >
< div class = "line" > < a id = "l01296" name = "l01296" > < / a > < span class = "lineno" > 1296< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01297" name = "l01297" > < / a > < span class = "lineno" > 1297< / span > < / div >
< div class = "line" > < a id = "l01299" name = "l01299" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga866af24e10db2797e1c5a5986dbf6c0d" > 1299< / 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 = "l01300" name = "l01300" > < / a > < span class = "lineno" > 1300< / span > < / div >
< div class = "line" > < a id = "l01302" name = "l01302" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga654fec16a9746b390916697a2ab2546e" > 1302< / 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 = "l01303" name = "l01303" > < / a > < span class = "lineno" > 1303< / span > < / div >
< div class = "line" > < a id = "l01305" name = "l01305" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga82a53e083205a965387b3c3e2463244a" > 1305< / 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 = "l01306" name = "l01306" > < / a > < span class = "lineno" > 1306< / span > < a class = "code hl_class" href = "classmlx_1_1core_1_1array.html" > array< / a > c,< / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l01307" name = "l01307" > < / a > < span class = "lineno" > 1307< / span > < a class = "code hl_class" href = "classmlx_1_1core_1_1array.html" > array< / a > a,< / div >
< div class = "line" > < a id = "l01308" name = "l01308" > < / a > < span class = "lineno" > 1308< / span > < a class = "code hl_class" href = "classmlx_1_1core_1_1array.html" > array< / a > b,< / div >
2024-07-26 02:59:11 +08:00
< div class = "line" > < a id = "l01309" name = "l01309" > < / a > < span class = "lineno" > 1309< / span > < span class = "keyword" > const< / span > < span class = "keywordtype" > float< / span > & alpha = 1.f,< / div >
< div class = "line" > < a id = "l01310" name = "l01310" > < / a > < span class = "lineno" > 1310< / span > < span class = "keyword" > const< / span > < span class = "keywordtype" > float< / span > & beta = 1.f,< / div >
< div class = "line" > < a id = "l01311" name = "l01311" > < / a > < span class = "lineno" > 1311< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01312" name = "l01312" > < / a > < span class = "lineno" > 1312< / span > < / div >
< div class = "line" > < a id = "l01314" name = "l01314" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga6b76c8ea46b19e6866af155fa5910be6" > 1314< / 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 = "l01315" name = "l01315" > < / a > < span class = "lineno" > 1315< / span > < a class = "code hl_class" href = "classmlx_1_1core_1_1array.html" > array< / a > a,< / div >
< div class = "line" > < a id = "l01316" name = "l01316" > < / a > < span class = "lineno" > 1316< / span > < a class = "code hl_class" href = "classmlx_1_1core_1_1array.html" > array< / a > b,< / div >
< div class = "line" > < a id = "l01317" name = "l01317" > < / a > < span class = "lineno" > 1317< / span > < span class = "keywordtype" > int< / span > block_size,< / div >
< div class = "line" > < a id = "l01318" name = "l01318" > < / a > < span class = "lineno" > 1318< / span > std::optional< array> mask_out = std::nullopt,< / div >
< div class = "line" > < a id = "l01319" name = "l01319" > < / a > < span class = "lineno" > 1319< / span > std::optional< array> mask_lhs = std::nullopt,< / div >
< div class = "line" > < a id = "l01320" name = "l01320" > < / a > < span class = "lineno" > 1320< / span > std::optional< array> mask_rhs = std::nullopt,< / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l01321" name = "l01321" > < / a > < span class = "lineno" > 1321< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01322" name = "l01322" > < / a > < span class = "lineno" > 1322< / span > < / div >
2024-07-26 02:59:11 +08:00
< div class = "line" > < a id = "l01324" name = "l01324" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga8d50480266d258cac40ff51bcb0fc6a7" > 1324< / 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 = "l01325" name = "l01325" > < / a > < span class = "lineno" > 1325< / span > < a class = "code hl_class" href = "classmlx_1_1core_1_1array.html" > array< / a > a,< / div >
< div class = "line" > < a id = "l01326" name = "l01326" > < / a > < span class = "lineno" > 1326< / span > < a class = "code hl_class" href = "classmlx_1_1core_1_1array.html" > array< / a > b,< / div >
< div class = "line" > < a id = "l01327" name = "l01327" > < / a > < span class = "lineno" > 1327< / span > std::optional< array> lhs_indices = std::nullopt,< / div >
< div class = "line" > < a id = "l01328" name = "l01328" > < / a > < span class = "lineno" > 1328< / span > std::optional< array> rhs_indices = std::nullopt,< / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l01329" name = "l01329" > < / a > < span class = "lineno" > 1329< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01330" name = "l01330" > < / a > < span class = "lineno" > 1330< / span > < / div >
2024-07-26 02:59:11 +08:00
< div class = "line" > < a id = "l01332" name = "l01332" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga9236b085a88ead3128ed8079d009cac6" > 1332< / 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 = "l01333" name = "l01333" > < / a > < span class = "lineno" > 1333< / 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 = "l01334" name = "l01334" > < / a > < span class = "lineno" > 1334< / span > < span class = "keywordtype" > int< / span > offset = 0,< / div >
< div class = "line" > < a id = "l01335" name = "l01335" > < / a > < span class = "lineno" > 1335< / span > < span class = "keywordtype" > int< / span > axis1 = 0,< / div >
< div class = "line" > < a id = "l01336" name = "l01336" > < / a > < span class = "lineno" > 1336< / span > < span class = "keywordtype" > int< / span > axis2 = 1,< / div >
< div class = "line" > < a id = "l01337" name = "l01337" > < / a > < span class = "lineno" > 1337< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01338" name = "l01338" > < / a > < span class = "lineno" > 1338< / span > < / div >
< div class = "line" > < a id = "l01340" name = "l01340" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga11af511875640e1fa88e0ca87e199344" > 1340< / 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 = "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#gabf786129c7660ed8d5acb5499bc6fefd" > 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#gabf786129c7660ed8d5acb5499bc6fefd" > trace< / a > (< / div >
< div class = "line" > < a id = "l01344" name = "l01344" > < / a > < span class = "lineno" > 1344< / 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 = "l01345" name = "l01345" > < / a > < span class = "lineno" > 1345< / span > < span class = "keywordtype" > int< / span > offset,< / div >
< div class = "line" > < a id = "l01346" name = "l01346" > < / a > < span class = "lineno" > 1346< / span > < span class = "keywordtype" > int< / span > axis1,< / div >
< div class = "line" > < a id = "l01347" name = "l01347" > < / a > < span class = "lineno" > 1347< / span > < span class = "keywordtype" > int< / span > axis2,< / div >
< div class = "line" > < a id = "l01348" name = "l01348" > < / a > < span class = "lineno" > 1348< / span > < a class = "code hl_struct" href = "structmlx_1_1core_1_1_dtype.html" > Dtype< / a > dtype,< / div >
< div class = "line" > < a id = "l01349" name = "l01349" > < / a > < span class = "lineno" > 1349< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01350" name = "l01350" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga5ed43c2dbf7d6cbddbaa2fd682deaafd" > 1350< / 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 = "l01351" name = "l01351" > < / a > < span class = "lineno" > 1351< / 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 = "l01352" name = "l01352" > < / a > < span class = "lineno" > 1352< / span > < span class = "keywordtype" > int< / span > offset,< / div >
< div class = "line" > < a id = "l01353" name = "l01353" > < / a > < span class = "lineno" > 1353< / span > < span class = "keywordtype" > int< / span > axis1,< / div >
< div class = "line" > < a id = "l01354" name = "l01354" > < / a > < span class = "lineno" > 1354< / span > < span class = "keywordtype" > int< / span > axis2,< / 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" > < a class = "line" href = "group__ops.html#gaf25c00108feaafaa6350a4434cb0062e" > 1356< / 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 = "l01357" name = "l01357" > < / a > < span class = "lineno" > 1357< / span > < / div >
< div class = "line" > < a id = "l01363" name = "l01363" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gac4a51a68fbe1725436b026d2fbb95759" > 1363< / a > < / span > std::vector< array> < a class = "code hl_function" href = "group__ops.html#gac4a51a68fbe1725436b026d2fbb95759" > depends< / a > (< / div >
< div class = "line" > < a id = "l01364" name = "l01364" > < / a > < span class = "lineno" > 1364< / span > < span class = "keyword" > const< / span > std::vector< array> & inputs,< / div >
< div class = "line" > < a id = "l01365" name = "l01365" > < / a > < span class = "lineno" > 1365< / span > < span class = "keyword" > const< / span > std::vector< array> & dependencies);< / 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#gaba4d25e7a2bf87ba4feb7837ec7fa396" > 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#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 = "l01369" name = "l01369" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga08ca172ce80157c916c89dd0b45b95c5" > 1369< / a > < / span > std::vector< array> < a class = "code hl_function" href = "group__ops.html#gaba4d25e7a2bf87ba4feb7837ec7fa396" > atleast_1d< / a > (< / div >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l01370" name = "l01370" > < / a > < span class = "lineno" > 1370< / span > < span class = "keyword" > const< / span > std::vector< array> & a,< / div >
< div class = "line" > < a id = "l01371" name = "l01371" > < / a > < span class = "lineno" > 1371< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
2024-07-26 02:59:11 +08:00
< div class = "line" > < a id = "l01372" name = "l01372" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaeeb7f5bb88aa32a3ac2be1f39c5f8087" > 1372< / 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 = "l01373" name = "l01373" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga9950299a80c2562f13448758f856d1f5" > 1373< / a > < / span > std::vector< array> < a class = "code hl_function" href = "group__ops.html#gaeeb7f5bb88aa32a3ac2be1f39c5f8087" > atleast_2d< / a > (< / div >
< div class = "line" > < a id = "l01374" name = "l01374" > < / a > < span class = "lineno" > 1374< / span > < span class = "keyword" > const< / span > std::vector< array> & a,< / div >
< div class = "line" > < a id = "l01375" name = "l01375" > < / a > < span class = "lineno" > 1375< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01376" name = "l01376" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga4afd919601e67782ff964465919956a0" > 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#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 = "l01377" name = "l01377" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaffdf742ad79440a60dda40062a8074fe" > 1377< / a > < / span > std::vector< array> < a class = "code hl_function" href = "group__ops.html#ga4afd919601e67782ff964465919956a0" > atleast_3d< / a > (< / div >
< div class = "line" > < a id = "l01378" name = "l01378" > < / a > < span class = "lineno" > 1378< / span > < span class = "keyword" > const< / span > std::vector< array> & a,< / div >
< div class = "line" > < a id = "l01379" name = "l01379" > < / a > < span class = "lineno" > 1379< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01380" name = "l01380" > < / a > < span class = "lineno" > 1380< / span > < / div >
< div class = "line" > < a id = "l01385" name = "l01385" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga6d5f5f72362488b956cdc4615ef6c636" > 1385< / 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 = "l01386" name = "l01386" > < / a > < span class = "lineno" > 1386< / 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 = "l01387" name = "l01387" > < / a > < span class = "lineno" > 1387< / span > std::vector< int> axes,< / div >
< div class = "line" > < a id = "l01388" name = "l01388" > < / a > < span class = "lineno" > 1388< / span > < span class = "keywordtype" > bool< / span > inverted,< / div >
< div class = "line" > < a id = "l01389" name = "l01389" > < / a > < span class = "lineno" > 1389< / 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 = "l01390" name = "l01390" > < / a > < span class = "lineno" > 1390< / span > < a class = "code hl_typedef" href = "namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58" > StreamOrDevice< / a > s = {});< / div >
< div class = "line" > < a id = "l01391" name = "l01391" > < / a > < span class = "lineno" > 1391< / span > < / div >
< div class = "line" > < a id = "l01392" name = "l01392" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga5b596906bf8cdc8d97ed6ddc9aeb4c23" > 1392< / 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 >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l01393" name = "l01393" > < / a > < span class = "lineno" > 1393< / span > < / div >
2024-07-26 02:59:11 +08:00
< div class = "line" > < a id = "l01395" name = "l01395" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga752fd2707dabb05d0308ba3d55346ada" > 1395< / 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 = "l01396" name = "l01396" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gaf0d232de4cbfffda1e2c838f8afdf6ff" > 1396< / 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 >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l01397" name = "l01397" > < / a > < span class = "lineno" > 1397< / span > < / div >
2024-07-26 02:59:11 +08:00
< div class = "line" > < a id = "l01399" name = "l01399" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga8af4f22c08c11c4ffab7e3d45e0f3cd6" > 1399< / 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 = "l01400" name = "l01400" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga52392a2a98f09a80da8d338c4908bd02" > 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#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 >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l01401" name = "l01401" > < / a > < span class = "lineno" > 1401< / span > < / div >
2024-07-26 02:59:11 +08:00
< div class = "line" > < a id = "l01403" name = "l01403" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga3188638fba3a60e264baf69956a1e08b" > 1403< / 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 = "l01404" name = "l01404" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gac3a6fe18694e84b3d63458e9553ac181" > 1404< / 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 >
2024-07-12 06:32:08 +08:00
< div class = "line" > < a id = "l01405" name = "l01405" > < / a > < span class = "lineno" > 1405< / span > < / div >
2024-07-26 02:59:11 +08:00
< div class = "line" > < a id = "l01407" name = "l01407" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga89682bf78491761e062d4ee7bef0c829" > 1407< / 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 = "l01408" name = "l01408" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gad656c30f9fd7d9467e405657b325aa7e" > 1408< / 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 = "l01409" name = "l01409" > < / a > < span class = "lineno" > 1409< / span > < / div >
< div class = "line" > < a id = "l01411" name = "l01411" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#gafa376ad57d38ba87378f0272dc379b23" > 1411< / 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 = "l01412" name = "l01412" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga498b61f7e8f056ae00297fa0dc17303a" > 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#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 = "l01413" name = "l01413" > < / a > < span class = "lineno" > 1413< / span > < / div >
< div class = "line" > < a id = "l01414" name = "l01414" > < / a > < span class = "lineno" > < a class = "line" href = "group__ops.html#ga3602aa91b7b124a0b41ec1b2137a1b02" > 1414< / 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 = "l01417" name = "l01417" > < / a > < span class = "lineno" > 1417< / span > } < span class = "comment" > // namespace mlx::core< / span > < / div >
2024-06-07 11:28:06 +08:00
< 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 >
2024-07-12 06:32:08 +08:00
< 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:343< / div > < / div >
2024-06-07 11:28:06 +08:00
< 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 >
2024-07-12 06:32:08 +08:00
< 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:371< / div > < / div >
2024-06-07 11:28:06 +08:00
< 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_ga4fd04d6472ae7cdb94a174edfe338ea3" > < div class = "ttname" > < a href = "group__ops.html#ga4fd04d6472ae7cdb94a174edfe338ea3" > 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), StreamOrDevice s={})< / div > < div class = "ttdoc" > Pad an array with a constant value.< / div > < / div >
2024-07-26 02:59:11 +08:00
< div class = "ttc" id = "agroup__ops_html_ga506d1b63e391e45a7d8dc10cfd71e78d" > < div class = "ttname" > < a href = "group__ops.html#ga506d1b63e391e45a7d8dc10cfd71e78d" > 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 >
2024-06-07 11:28:06 +08:00
< 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_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 >
2024-07-12 06:32:08 +08:00
< 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:329< / div > < / div >
2024-06-07 11:28:06 +08:00
< 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_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 >
2024-07-26 02:59:11 +08:00
< 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 >
2024-06-07 11:28:06 +08:00
< 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_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_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_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_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_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_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 >
2024-07-12 06:32:08 +08:00
< 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:357< / div > < / div >
2024-06-07 11:28:06 +08:00
< 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_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:69< / 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:73< / 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:15< / 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.10.0
< / small > < / address >
< / body >
< / html >
