| <!DOCTYPE html><html lang="en"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><meta name="generator" content="rustdoc"><meta name="description" content="This module implements the `NDArray` type for working with TVM tensors or coverting from a Rust’s ndarray to TVM `NDArray`."><title>tvm_rt::ndarray - Rust</title><link rel="preload" as="font" type="font/woff2" crossorigin href="../../static.files/SourceSerif4-Regular-46f98efaafac5295.ttf.woff2"><link rel="preload" as="font" type="font/woff2" crossorigin href="../../static.files/FiraSans-Regular-018c141bf0843ffd.woff2"><link rel="preload" as="font" type="font/woff2" crossorigin href="../../static.files/FiraSans-Medium-8f9a781e4970d388.woff2"><link rel="preload" as="font" type="font/woff2" crossorigin href="../../static.files/SourceCodePro-Regular-562dcc5011b6de7d.ttf.woff2"><link rel="preload" as="font" type="font/woff2" crossorigin href="../../static.files/SourceSerif4-Bold-a2c9cd1067f8b328.ttf.woff2"><link rel="preload" as="font" type="font/woff2" crossorigin href="../../static.files/SourceCodePro-Semibold-d899c5a5c4aeb14a.ttf.woff2"><link rel="stylesheet" href="../../static.files/normalize-76eba96aa4d2e634.css"><link rel="stylesheet" href="../../static.files/rustdoc-c4dbdcde0fbd8430.css" id="mainThemeStyle"><link rel="stylesheet" id="themeStyle" href="../../static.files/light-db279b6232be9c13.css"><link rel="stylesheet" disabled href="../../static.files/dark-cf923f49f397b216.css"><link rel="stylesheet" disabled href="../../static.files/ayu-be46fdc453a55015.css"><script src="../../static.files/storage-3891ce972e3a2bf8.js"></script><script defer src="../../static.files/main-98a684e84ae5b08b.js"></script><noscript><link rel="stylesheet" href="../../static.files/noscript-13285aec31fa243e.css"></noscript><link rel="alternate icon" type="image/png" href="../../static.files/favicon-16x16-8b506e7a72182f1c.png"><link rel="alternate icon" type="image/png" href="../../static.files/favicon-32x32-422f7d1d52889060.png"><link rel="icon" type="image/svg+xml" href="../../static.files/favicon-2c020d218678b618.svg"></head><body class="rustdoc mod"><!--[if lte IE 11]><div class="warning">This old browser is unsupported and will most likely display funky things.</div><![endif]--><nav class="mobile-topbar"><button class="sidebar-menu-toggle">☰</button><a class="logo-container" href="../../tvm_rt/index.html"><img class="rust-logo" src="../../static.files/rust-logo-151179464ae7ed46.svg" alt="logo"></a><h2></h2></nav><nav class="sidebar"><a class="logo-container" href="../../tvm_rt/index.html"><img class="rust-logo" src="../../static.files/rust-logo-151179464ae7ed46.svg" alt="logo"></a><h2 class="location"><a href="#">Module ndarray</a></h2><div class="sidebar-elems"><section><ul class="block"><li><a href="#structs">Structs</a></li><li><a href="#traits">Traits</a></li></ul></section></div></nav><main><div class="width-limiter"><nav class="sub"><form class="search-form"><span></span><input class="search-input" name="search" aria-label="Run search in the documentation" autocomplete="off" spellcheck="false" placeholder="Click or press ‘S’ to search, ‘?’ for more options…" type="search"><div id="help-button" title="help" tabindex="-1"><a href="../../help.html">?</a></div><div id="settings-menu" tabindex="-1"><a href="../../settings.html" title="settings"><img width="22" height="22" alt="Change settings" src="../../static.files/wheel-7b819b6101059cd0.svg"></a></div></form></nav><section id="main-content" class="content"><div class="main-heading"><h1>Module <a href="../index.html">tvm_rt</a>::<wbr><a class="mod" href="#">ndarray</a><button id="copy-path" title="Copy item path to clipboard"><img src="../../static.files/clipboard-7571035ce49a181d.svg" width="19" height="18" alt="Copy item path"></button></h1><span class="out-of-band"><a class="srclink" href="../../src/tvm_rt/ndarray.rs.html#20-515">source</a> · <button id="toggle-all-docs" title="collapse all docs">[<span>−</span>]</button></span></div><details class="toggle top-doc" open><summary class="hideme"><span>Expand description</span></summary><div class="docblock"><p>This module implements the <a href="struct.NDArray.html" title="struct tvm_rt::ndarray::NDArray"><code>NDArray</code></a> type for working with <em>TVM tensors</em> or |
| coverting from a Rust’s ndarray to TVM <code>NDArray</code>.</p> |
| <p>One can create an empty NDArray given the shape, device and dtype using [<code>empty</code>]. |
| To create an NDArray from a mutable buffer in cpu use <a href="struct.NDArray.html#method.copy_from_buffer"><code>copy_from_buffer</code></a>. |
| To copy an NDArray to different device use <a href="struct.NDArray.html#method.copy_to_device"><code>copy_to_device</code></a>.</p> |
| <p>Given a <a href="https://docs.rs/ndarray/0.12.1/ndarray/"><code>Rust's dynamic ndarray</code></a>, one can convert it to TVM NDArray as follows:</p> |
| <h2 id="example"><a href="#example">Example</a></h2> |
| <div class="example-wrap"><pre class="rust rust-example-rendered"><code><span class="kw">use </span>std::convert::TryFrom; |
| |
| <span class="kw">let </span>a = Array::from_shape_vec((<span class="number">2</span>, <span class="number">2</span>), <span class="macro">vec!</span>[<span class="number">1f32</span>, <span class="number">2.</span>, <span class="number">3.</span>, <span class="number">4.</span>]) |
| .unwrap() |
| .into_dyn(); <span class="comment">// Rust's ndarray |
| </span><span class="kw">let </span>nd = NDArray::from_rust_ndarray(<span class="kw-2">&</span>a, Device::cpu(<span class="number">0</span>), DataType::from_str(<span class="string">"float32"</span>).unwrap()).unwrap(); |
| <span class="macro">assert_eq!</span>(nd.shape(), <span class="kw-2">&</span>[<span class="number">2</span>, <span class="number">2</span>]); |
| <span class="kw">let </span>rnd: ArrayD<f32> = ArrayD::try_from(<span class="kw-2">&</span>nd).unwrap(); |
| <span class="macro">assert!</span>(rnd.all_close(<span class="kw-2">&</span>a, <span class="number">1e-8f32</span>));</code></pre></div> |
| </div></details><h2 id="structs" class="small-section-header"><a href="#structs">Structs</a></h2><ul class="item-table"><li><div class="item-name"><a class="struct" href="struct.NDArray.html" title="struct tvm_rt::ndarray::NDArray">NDArray</a></div></li><li><div class="item-name"><a class="struct" href="struct.NDArrayContainer.html" title="struct tvm_rt::ndarray::NDArrayContainer">NDArrayContainer</a></div><div class="desc docblock-short">See the <a href="../ndarray/index.html"><code>module-level documentation</code></a> for more details.</div></li></ul><h2 id="traits" class="small-section-header"><a href="#traits">Traits</a></h2><ul class="item-table"><li><div class="item-name"><a class="trait" href="trait.Num32.html" title="trait tvm_rt::ndarray::Num32">Num32</a></div><div class="desc docblock-short">A trait for the supported 32-bits numerical types in frontend.</div></li></ul></section></div></main><div id="rustdoc-vars" data-root-path="../../" data-static-root-path="../../static.files/" data-current-crate="tvm_rt" data-themes="" data-resource-suffix="" data-rustdoc-version="1.69.0 (84c898d65 2023-04-16)" data-search-js="search-8a59a8356673ec1f.js" data-settings-js="settings-f0c5c39777a9a2f6.js" data-settings-css="settings-0bcba95ff279c1db.css" ></div></body></html> |
