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

docs up

Browse Source
This commit is contained in:
Awni Hannun
2024-09-17 12:06:14 -07:00
committed by CircleCI Docs
parent 27e07888b0
commit 6a6ffb598d
739 changed files with 28107 additions and 8524 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/build/html/transforms_8h_source.html vendored
View File

@@ -255,7 +255,7 @@ $(function() { codefold.init(0); });
<div class="ttc" id="anamespacemlx_1_1core_html_a933289d4688479e1c4d8ba04332c406b"><div class="ttname"><a href="namespacemlx_1_1core.html#a933289d4688479e1c4d8ba04332c406b">mlx::core::vmap</a></div><div class="ttdeci">std::function&lt; array(const array &amp;) vmap)(const std::function&lt; array(const array &amp;)&gt; &amp;fun, int in_axis=0, int out_axis=0)</div><div class="ttdoc">Automatically vectorize a unary function over the requested axes.</div></div>
<div class="ttc" id="anamespacemlx_1_1core_html_a99994806f81e83f98746c9e5ba4c3673"><div class="ttname"><a href="namespacemlx_1_1core.html#a99994806f81e83f98746c9e5ba4c3673">mlx::core::grad</a></div><div class="ttdeci">std::function&lt; std::vector&lt; array &gt;(const std::vector&lt; array &gt; &amp;) grad)(const std::function&lt; array(const std::vector&lt; array &gt; &amp;)&gt; &amp;fun, const std::vector&lt; int &gt; &amp;argnums)</div><div class="ttdoc">Returns a function which computes the gradient of the input function with respect to a vector of inpu...</div><div class="ttdef"><b>Definition</b> transforms.h:122</div></div>
<div class="ttc" id="anamespacemlx_1_1core_html_ab79d66ddf1ec38b2f2c01234892a2230"><div class="ttname"><a href="namespacemlx_1_1core.html#ab79d66ddf1ec38b2f2c01234892a2230">mlx::core::ValueAndGradFn</a></div><div class="ttdeci">std::function&lt; std::pair&lt; std::vector&lt; array &gt;, std::vector&lt; array &gt; &gt;( const std::vector&lt; array &gt; &amp;)&gt; ValueAndGradFn</div><div class="ttdef"><b>Definition</b> transforms.h:63</div></div>
<div class="ttc" id="anamespacemlx_1_1core_html_af89751d79339f3e4d9318ea97d64d114"><div class="ttname"><a href="namespacemlx_1_1core.html#af89751d79339f3e4d9318ea97d64d114">mlx::core::enable_for_arrays_t</a></div><div class="ttdeci">typename std::enable_if_t&lt; is_arrays_v&lt; T... &gt; &gt; enable_for_arrays_t</div><div class="ttdef"><b>Definition</b> array.h:566</div></div>
<div class="ttc" id="anamespacemlx_1_1core_html_af89751d79339f3e4d9318ea97d64d114"><div class="ttname"><a href="namespacemlx_1_1core.html#af89751d79339f3e4d9318ea97d64d114">mlx::core::enable_for_arrays_t</a></div><div class="ttdeci">typename std::enable_if_t&lt; is_arrays_v&lt; T... &gt; &gt; enable_for_arrays_t</div><div class="ttdef"><b>Definition</b> array.h:589</div></div>
</div><!-- fragment --></div><!-- contents -->
<!-- start footer part -->
<hr class="footer"/><address class="footer"><small>