mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-25 01:41:17 +08:00
3511 lines
815 KiB
HTML
3511 lines
815 KiB
HTML
|
|
<!DOCTYPE html>
|
|
|
|
|
|
<html lang="en" data-content_root="../" >
|
|
|
|
<head>
|
|
<meta charset="utf-8" />
|
|
<meta name="viewport" content="width=device-width, initial-scale=1.0" /><meta name="viewport" content="width=device-width, initial-scale=1" />
|
|
|
|
<title>Operations — MLX 0.26.1 documentation</title>
|
|
|
|
|
|
|
|
<script data-cfasync="false">
|
|
document.documentElement.dataset.mode = localStorage.getItem("mode") || "";
|
|
document.documentElement.dataset.theme = localStorage.getItem("theme") || "";
|
|
</script>
|
|
|
|
<!-- Loaded before other Sphinx assets -->
|
|
<link href="../_static/styles/theme.css?digest=dfe6caa3a7d634c4db9b" rel="stylesheet" />
|
|
<link href="../_static/styles/bootstrap.css?digest=dfe6caa3a7d634c4db9b" rel="stylesheet" />
|
|
<link href="../_static/styles/pydata-sphinx-theme.css?digest=dfe6caa3a7d634c4db9b" rel="stylesheet" />
|
|
|
|
|
|
<link href="../_static/vendor/fontawesome/6.5.2/css/all.min.css?digest=dfe6caa3a7d634c4db9b" rel="stylesheet" />
|
|
<link rel="preload" as="font" type="font/woff2" crossorigin href="../_static/vendor/fontawesome/6.5.2/webfonts/fa-solid-900.woff2" />
|
|
<link rel="preload" as="font" type="font/woff2" crossorigin href="../_static/vendor/fontawesome/6.5.2/webfonts/fa-brands-400.woff2" />
|
|
<link rel="preload" as="font" type="font/woff2" crossorigin href="../_static/vendor/fontawesome/6.5.2/webfonts/fa-regular-400.woff2" />
|
|
|
|
<link rel="stylesheet" type="text/css" href="../_static/pygments.css?v=03e43079" />
|
|
<link rel="stylesheet" type="text/css" href="../_static/styles/sphinx-book-theme.css?v=eba8b062" />
|
|
|
|
<!-- Pre-loaded scripts that we'll load fully later -->
|
|
<link rel="preload" as="script" href="../_static/scripts/bootstrap.js?digest=dfe6caa3a7d634c4db9b" />
|
|
<link rel="preload" as="script" href="../_static/scripts/pydata-sphinx-theme.js?digest=dfe6caa3a7d634c4db9b" />
|
|
<script src="../_static/vendor/fontawesome/6.5.2/js/all.min.js?digest=dfe6caa3a7d634c4db9b"></script>
|
|
|
|
<script src="../_static/documentation_options.js?v=3724ff34"></script>
|
|
<script src="../_static/doctools.js?v=9a2dae69"></script>
|
|
<script src="../_static/sphinx_highlight.js?v=dc90522c"></script>
|
|
<script src="../_static/scripts/sphinx-book-theme.js?v=887ef09a"></script>
|
|
<script>DOCUMENTATION_OPTIONS.pagename = 'cpp/ops';</script>
|
|
<link rel="icon" href="../_static/mlx_logo.png"/>
|
|
<link rel="index" title="Index" href="../genindex.html" />
|
|
<link rel="search" title="Search" href="../search.html" />
|
|
<link rel="next" title="Custom Extensions in MLX" href="../dev/extensions.html" />
|
|
<link rel="prev" title="mlx.utils.tree_reduce" href="../python/_autosummary/mlx.utils.tree_reduce.html" />
|
|
<meta name="viewport" content="width=device-width, initial-scale=1"/>
|
|
<meta name="docsearch:language" content="en"/>
|
|
</head>
|
|
|
|
|
|
<body data-bs-spy="scroll" data-bs-target=".bd-toc-nav" data-offset="180" data-bs-root-margin="0px 0px -60%" data-default-mode="">
|
|
|
|
|
|
|
|
<div id="pst-skip-link" class="skip-link d-print-none"><a href="#main-content">Skip to main content</a></div>
|
|
|
|
<div id="pst-scroll-pixel-helper"></div>
|
|
|
|
<button type="button" class="btn rounded-pill" id="pst-back-to-top">
|
|
<i class="fa-solid fa-arrow-up"></i>Back to top</button>
|
|
|
|
|
|
<input type="checkbox"
|
|
class="sidebar-toggle"
|
|
id="pst-primary-sidebar-checkbox"/>
|
|
<label class="overlay overlay-primary" for="pst-primary-sidebar-checkbox"></label>
|
|
|
|
<input type="checkbox"
|
|
class="sidebar-toggle"
|
|
id="pst-secondary-sidebar-checkbox"/>
|
|
<label class="overlay overlay-secondary" for="pst-secondary-sidebar-checkbox"></label>
|
|
|
|
<div class="search-button__wrapper">
|
|
<div class="search-button__overlay"></div>
|
|
<div class="search-button__search-container">
|
|
<form class="bd-search d-flex align-items-center"
|
|
action="../search.html"
|
|
method="get">
|
|
<i class="fa-solid fa-magnifying-glass"></i>
|
|
<input type="search"
|
|
class="form-control"
|
|
name="q"
|
|
id="search-input"
|
|
placeholder="Search..."
|
|
aria-label="Search..."
|
|
autocomplete="off"
|
|
autocorrect="off"
|
|
autocapitalize="off"
|
|
spellcheck="false"/>
|
|
<span class="search-button__kbd-shortcut"><kbd class="kbd-shortcut__modifier">Ctrl</kbd>+<kbd>K</kbd></span>
|
|
</form></div>
|
|
</div>
|
|
|
|
<div class="pst-async-banner-revealer d-none">
|
|
<aside id="bd-header-version-warning" class="d-none d-print-none" aria-label="Version warning"></aside>
|
|
</div>
|
|
|
|
|
|
<header class="bd-header navbar navbar-expand-lg bd-navbar d-print-none">
|
|
</header>
|
|
|
|
|
|
<div class="bd-container">
|
|
<div class="bd-container__inner bd-page-width">
|
|
|
|
|
|
|
|
<div class="bd-sidebar-primary bd-sidebar">
|
|
|
|
|
|
|
|
<div class="sidebar-header-items sidebar-primary__section">
|
|
|
|
|
|
|
|
|
|
</div>
|
|
|
|
<div class="sidebar-primary-items__start sidebar-primary__section">
|
|
<div class="sidebar-primary-item">
|
|
|
|
|
|
|
|
|
|
|
|
<a class="navbar-brand logo" href="../index.html">
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
<img src="../_static/mlx_logo.png" class="logo__image only-light" alt="MLX 0.26.1 documentation - Home"/>
|
|
<script>document.write(`<img src="../_static/mlx_logo_dark.png" class="logo__image only-dark" alt="MLX 0.26.1 documentation - Home"/>`);</script>
|
|
|
|
|
|
</a></div>
|
|
<div class="sidebar-primary-item">
|
|
|
|
<script>
|
|
document.write(`
|
|
<button class="btn search-button-field search-button__button" title="Search" aria-label="Search" data-bs-placement="bottom" data-bs-toggle="tooltip">
|
|
<i class="fa-solid fa-magnifying-glass"></i>
|
|
<span class="search-button__default-text">Search</span>
|
|
<span class="search-button__kbd-shortcut"><kbd class="kbd-shortcut__modifier">Ctrl</kbd>+<kbd class="kbd-shortcut__modifier">K</kbd></span>
|
|
</button>
|
|
`);
|
|
</script></div>
|
|
<div class="sidebar-primary-item"><nav class="bd-links bd-docs-nav" aria-label="Main">
|
|
<div class="bd-toc-item navbar-nav active">
|
|
<p aria-level="2" class="caption" role="heading"><span class="caption-text">Install</span></p>
|
|
<ul class="nav bd-sidenav">
|
|
<li class="toctree-l1"><a class="reference internal" href="../install.html">Build and Install</a></li>
|
|
</ul>
|
|
<p aria-level="2" class="caption" role="heading"><span class="caption-text">Usage</span></p>
|
|
<ul class="nav bd-sidenav">
|
|
<li class="toctree-l1"><a class="reference internal" href="../usage/quick_start.html">Quick Start Guide</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../usage/lazy_evaluation.html">Lazy Evaluation</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../usage/unified_memory.html">Unified Memory</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../usage/indexing.html">Indexing Arrays</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../usage/saving_and_loading.html">Saving and Loading Arrays</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../usage/function_transforms.html">Function Transforms</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../usage/compile.html">Compilation</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../usage/numpy.html">Conversion to NumPy and Other Frameworks</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../usage/distributed.html">Distributed Communication</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../usage/using_streams.html">Using Streams</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../usage/export.html">Exporting Functions</a></li>
|
|
</ul>
|
|
<p aria-level="2" class="caption" role="heading"><span class="caption-text">Examples</span></p>
|
|
<ul class="nav bd-sidenav">
|
|
<li class="toctree-l1"><a class="reference internal" href="../examples/linear_regression.html">Linear Regression</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../examples/mlp.html">Multi-Layer Perceptron</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../examples/llama-inference.html">LLM inference</a></li>
|
|
</ul>
|
|
<p aria-level="2" class="caption" role="heading"><span class="caption-text">Python API Reference</span></p>
|
|
<ul class="nav bd-sidenav">
|
|
<li class="toctree-l1 has-children"><a class="reference internal" href="../python/array.html">Array</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.html">mlx.core.array</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.astype.html">mlx.core.array.astype</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.at.html">mlx.core.array.at</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.item.html">mlx.core.array.item</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.tolist.html">mlx.core.array.tolist</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.dtype.html">mlx.core.array.dtype</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.itemsize.html">mlx.core.array.itemsize</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.nbytes.html">mlx.core.array.nbytes</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.ndim.html">mlx.core.array.ndim</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.shape.html">mlx.core.array.shape</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.size.html">mlx.core.array.size</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.real.html">mlx.core.array.real</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.imag.html">mlx.core.array.imag</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.abs.html">mlx.core.array.abs</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.all.html">mlx.core.array.all</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.any.html">mlx.core.array.any</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.argmax.html">mlx.core.array.argmax</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.argmin.html">mlx.core.array.argmin</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.conj.html">mlx.core.array.conj</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.cos.html">mlx.core.array.cos</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.cummax.html">mlx.core.array.cummax</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.cummin.html">mlx.core.array.cummin</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.cumprod.html">mlx.core.array.cumprod</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.cumsum.html">mlx.core.array.cumsum</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.diag.html">mlx.core.array.diag</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.diagonal.html">mlx.core.array.diagonal</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.exp.html">mlx.core.array.exp</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.flatten.html">mlx.core.array.flatten</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.log.html">mlx.core.array.log</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.log10.html">mlx.core.array.log10</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.log1p.html">mlx.core.array.log1p</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.log2.html">mlx.core.array.log2</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.logcumsumexp.html">mlx.core.array.logcumsumexp</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.logsumexp.html">mlx.core.array.logsumexp</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.max.html">mlx.core.array.max</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.mean.html">mlx.core.array.mean</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.min.html">mlx.core.array.min</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.moveaxis.html">mlx.core.array.moveaxis</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.prod.html">mlx.core.array.prod</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.reciprocal.html">mlx.core.array.reciprocal</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.reshape.html">mlx.core.array.reshape</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.round.html">mlx.core.array.round</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.rsqrt.html">mlx.core.array.rsqrt</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.sin.html">mlx.core.array.sin</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.split.html">mlx.core.array.split</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.sqrt.html">mlx.core.array.sqrt</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.square.html">mlx.core.array.square</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.squeeze.html">mlx.core.array.squeeze</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.std.html">mlx.core.array.std</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.sum.html">mlx.core.array.sum</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.swapaxes.html">mlx.core.array.swapaxes</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.transpose.html">mlx.core.array.transpose</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.T.html">mlx.core.array.T</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.var.html">mlx.core.array.var</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array.view.html">mlx.core.array.view</a></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l1 has-children"><a class="reference internal" href="../python/data_types.html">Data Types</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.Dtype.html">mlx.core.Dtype</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.DtypeCategory.html">mlx.core.DtypeCategory</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.issubdtype.html">mlx.core.issubdtype</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.finfo.html">mlx.core.finfo</a></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l1 has-children"><a class="reference internal" href="../python/devices_and_streams.html">Devices and Streams</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.Device.html">mlx.core.Device</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/stream_class.html">mlx.core.Stream</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.default_device.html">mlx.core.default_device</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.set_default_device.html">mlx.core.set_default_device</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.default_stream.html">mlx.core.default_stream</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.new_stream.html">mlx.core.new_stream</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.set_default_stream.html">mlx.core.set_default_stream</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.stream.html">mlx.core.stream</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.synchronize.html">mlx.core.synchronize</a></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l1 has-children"><a class="reference internal" href="../python/export.html">Export Functions</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.export_function.html">mlx.core.export_function</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.import_function.html">mlx.core.import_function</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.exporter.html">mlx.core.exporter</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.export_to_dot.html">mlx.core.export_to_dot</a></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l1 has-children"><a class="reference internal" href="../python/ops.html">Operations</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.abs.html">mlx.core.abs</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.add.html">mlx.core.add</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.addmm.html">mlx.core.addmm</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.all.html">mlx.core.all</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.allclose.html">mlx.core.allclose</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.any.html">mlx.core.any</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.arange.html">mlx.core.arange</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.arccos.html">mlx.core.arccos</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.arccosh.html">mlx.core.arccosh</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.arcsin.html">mlx.core.arcsin</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.arcsinh.html">mlx.core.arcsinh</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.arctan.html">mlx.core.arctan</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.arctan2.html">mlx.core.arctan2</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.arctanh.html">mlx.core.arctanh</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.argmax.html">mlx.core.argmax</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.argmin.html">mlx.core.argmin</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.argpartition.html">mlx.core.argpartition</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.argsort.html">mlx.core.argsort</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.array_equal.html">mlx.core.array_equal</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.as_strided.html">mlx.core.as_strided</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.atleast_1d.html">mlx.core.atleast_1d</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.atleast_2d.html">mlx.core.atleast_2d</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.atleast_3d.html">mlx.core.atleast_3d</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.bitwise_and.html">mlx.core.bitwise_and</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.bitwise_invert.html">mlx.core.bitwise_invert</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.bitwise_or.html">mlx.core.bitwise_or</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.bitwise_xor.html">mlx.core.bitwise_xor</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.block_masked_mm.html">mlx.core.block_masked_mm</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.broadcast_arrays.html">mlx.core.broadcast_arrays</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.broadcast_to.html">mlx.core.broadcast_to</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.ceil.html">mlx.core.ceil</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.clip.html">mlx.core.clip</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.concatenate.html">mlx.core.concatenate</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.contiguous.html">mlx.core.contiguous</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.conj.html">mlx.core.conj</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.conjugate.html">mlx.core.conjugate</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.convolve.html">mlx.core.convolve</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.conv1d.html">mlx.core.conv1d</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.conv2d.html">mlx.core.conv2d</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.conv3d.html">mlx.core.conv3d</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.conv_transpose1d.html">mlx.core.conv_transpose1d</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.conv_transpose2d.html">mlx.core.conv_transpose2d</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.conv_transpose3d.html">mlx.core.conv_transpose3d</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.conv_general.html">mlx.core.conv_general</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.cos.html">mlx.core.cos</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.cosh.html">mlx.core.cosh</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.cummax.html">mlx.core.cummax</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.cummin.html">mlx.core.cummin</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.cumprod.html">mlx.core.cumprod</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.cumsum.html">mlx.core.cumsum</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.degrees.html">mlx.core.degrees</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.dequantize.html">mlx.core.dequantize</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.diag.html">mlx.core.diag</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.diagonal.html">mlx.core.diagonal</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.divide.html">mlx.core.divide</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.divmod.html">mlx.core.divmod</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.einsum.html">mlx.core.einsum</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.einsum_path.html">mlx.core.einsum_path</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.equal.html">mlx.core.equal</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.erf.html">mlx.core.erf</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.erfinv.html">mlx.core.erfinv</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.exp.html">mlx.core.exp</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.expm1.html">mlx.core.expm1</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.expand_dims.html">mlx.core.expand_dims</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.eye.html">mlx.core.eye</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.flatten.html">mlx.core.flatten</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.floor.html">mlx.core.floor</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.floor_divide.html">mlx.core.floor_divide</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.full.html">mlx.core.full</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.gather_mm.html">mlx.core.gather_mm</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.gather_qmm.html">mlx.core.gather_qmm</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.greater.html">mlx.core.greater</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.greater_equal.html">mlx.core.greater_equal</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.hadamard_transform.html">mlx.core.hadamard_transform</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.identity.html">mlx.core.identity</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.imag.html">mlx.core.imag</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.inner.html">mlx.core.inner</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.isfinite.html">mlx.core.isfinite</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.isclose.html">mlx.core.isclose</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.isinf.html">mlx.core.isinf</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.isnan.html">mlx.core.isnan</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.isneginf.html">mlx.core.isneginf</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.isposinf.html">mlx.core.isposinf</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.issubdtype.html">mlx.core.issubdtype</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.kron.html">mlx.core.kron</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.left_shift.html">mlx.core.left_shift</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.less.html">mlx.core.less</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.less_equal.html">mlx.core.less_equal</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.linspace.html">mlx.core.linspace</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.load.html">mlx.core.load</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.log.html">mlx.core.log</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.log2.html">mlx.core.log2</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.log10.html">mlx.core.log10</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.log1p.html">mlx.core.log1p</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.logaddexp.html">mlx.core.logaddexp</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.logcumsumexp.html">mlx.core.logcumsumexp</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.logical_not.html">mlx.core.logical_not</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.logical_and.html">mlx.core.logical_and</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.logical_or.html">mlx.core.logical_or</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.logsumexp.html">mlx.core.logsumexp</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.matmul.html">mlx.core.matmul</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.max.html">mlx.core.max</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.maximum.html">mlx.core.maximum</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.mean.html">mlx.core.mean</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.meshgrid.html">mlx.core.meshgrid</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.min.html">mlx.core.min</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.minimum.html">mlx.core.minimum</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.moveaxis.html">mlx.core.moveaxis</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.multiply.html">mlx.core.multiply</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.nan_to_num.html">mlx.core.nan_to_num</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.negative.html">mlx.core.negative</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.not_equal.html">mlx.core.not_equal</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.ones.html">mlx.core.ones</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.ones_like.html">mlx.core.ones_like</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.outer.html">mlx.core.outer</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.partition.html">mlx.core.partition</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.pad.html">mlx.core.pad</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.power.html">mlx.core.power</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.prod.html">mlx.core.prod</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.put_along_axis.html">mlx.core.put_along_axis</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.quantize.html">mlx.core.quantize</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.quantized_matmul.html">mlx.core.quantized_matmul</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.radians.html">mlx.core.radians</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.real.html">mlx.core.real</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.reciprocal.html">mlx.core.reciprocal</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.remainder.html">mlx.core.remainder</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.repeat.html">mlx.core.repeat</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.reshape.html">mlx.core.reshape</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.right_shift.html">mlx.core.right_shift</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.roll.html">mlx.core.roll</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.round.html">mlx.core.round</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.rsqrt.html">mlx.core.rsqrt</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.save.html">mlx.core.save</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.savez.html">mlx.core.savez</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.savez_compressed.html">mlx.core.savez_compressed</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.save_gguf.html">mlx.core.save_gguf</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.save_safetensors.html">mlx.core.save_safetensors</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.sigmoid.html">mlx.core.sigmoid</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.sign.html">mlx.core.sign</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.sin.html">mlx.core.sin</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.sinh.html">mlx.core.sinh</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.slice.html">mlx.core.slice</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.slice_update.html">mlx.core.slice_update</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.softmax.html">mlx.core.softmax</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.sort.html">mlx.core.sort</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.split.html">mlx.core.split</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.sqrt.html">mlx.core.sqrt</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.square.html">mlx.core.square</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.squeeze.html">mlx.core.squeeze</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.stack.html">mlx.core.stack</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.std.html">mlx.core.std</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.stop_gradient.html">mlx.core.stop_gradient</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.subtract.html">mlx.core.subtract</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.sum.html">mlx.core.sum</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.swapaxes.html">mlx.core.swapaxes</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.take.html">mlx.core.take</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.take_along_axis.html">mlx.core.take_along_axis</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.tan.html">mlx.core.tan</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.tanh.html">mlx.core.tanh</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.tensordot.html">mlx.core.tensordot</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.tile.html">mlx.core.tile</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.topk.html">mlx.core.topk</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.trace.html">mlx.core.trace</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.transpose.html">mlx.core.transpose</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.tri.html">mlx.core.tri</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.tril.html">mlx.core.tril</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.triu.html">mlx.core.triu</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.unflatten.html">mlx.core.unflatten</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.var.html">mlx.core.var</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.view.html">mlx.core.view</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.where.html">mlx.core.where</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.zeros.html">mlx.core.zeros</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.zeros_like.html">mlx.core.zeros_like</a></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l1 has-children"><a class="reference internal" href="../python/random.html">Random</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.random.bernoulli.html">mlx.core.random.bernoulli</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.random.categorical.html">mlx.core.random.categorical</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.random.gumbel.html">mlx.core.random.gumbel</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.random.key.html">mlx.core.random.key</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.random.normal.html">mlx.core.random.normal</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.random.multivariate_normal.html">mlx.core.random.multivariate_normal</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.random.randint.html">mlx.core.random.randint</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.random.seed.html">mlx.core.random.seed</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.random.split.html">mlx.core.random.split</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.random.truncated_normal.html">mlx.core.random.truncated_normal</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.random.uniform.html">mlx.core.random.uniform</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.random.laplace.html">mlx.core.random.laplace</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.random.permutation.html">mlx.core.random.permutation</a></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l1 has-children"><a class="reference internal" href="../python/transforms.html">Transforms</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.eval.html">mlx.core.eval</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.async_eval.html">mlx.core.async_eval</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.compile.html">mlx.core.compile</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.custom_function.html">mlx.core.custom_function</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.disable_compile.html">mlx.core.disable_compile</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.enable_compile.html">mlx.core.enable_compile</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.grad.html">mlx.core.grad</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.value_and_grad.html">mlx.core.value_and_grad</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.jvp.html">mlx.core.jvp</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.vjp.html">mlx.core.vjp</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.vmap.html">mlx.core.vmap</a></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l1 has-children"><a class="reference internal" href="../python/fast.html">Fast</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.fast.rms_norm.html">mlx.core.fast.rms_norm</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.fast.layer_norm.html">mlx.core.fast.layer_norm</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.fast.rope.html">mlx.core.fast.rope</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.fast.scaled_dot_product_attention.html">mlx.core.fast.scaled_dot_product_attention</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.fast.metal_kernel.html">mlx.core.fast.metal_kernel</a></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l1 has-children"><a class="reference internal" href="../python/fft.html">FFT</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.fft.fft.html">mlx.core.fft.fft</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.fft.ifft.html">mlx.core.fft.ifft</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.fft.fft2.html">mlx.core.fft.fft2</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.fft.ifft2.html">mlx.core.fft.ifft2</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.fft.fftn.html">mlx.core.fft.fftn</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.fft.ifftn.html">mlx.core.fft.ifftn</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.fft.rfft.html">mlx.core.fft.rfft</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.fft.irfft.html">mlx.core.fft.irfft</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.fft.rfft2.html">mlx.core.fft.rfft2</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.fft.irfft2.html">mlx.core.fft.irfft2</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.fft.rfftn.html">mlx.core.fft.rfftn</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.fft.irfftn.html">mlx.core.fft.irfftn</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.fft.fftshift.html">mlx.core.fft.fftshift</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.fft.ifftshift.html">mlx.core.fft.ifftshift</a></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l1 has-children"><a class="reference internal" href="../python/linalg.html">Linear Algebra</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.linalg.inv.html">mlx.core.linalg.inv</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.linalg.tri_inv.html">mlx.core.linalg.tri_inv</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.linalg.norm.html">mlx.core.linalg.norm</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.linalg.cholesky.html">mlx.core.linalg.cholesky</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.linalg.cholesky_inv.html">mlx.core.linalg.cholesky_inv</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.linalg.cross.html">mlx.core.linalg.cross</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.linalg.qr.html">mlx.core.linalg.qr</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.linalg.svd.html">mlx.core.linalg.svd</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.linalg.eigvals.html">mlx.core.linalg.eigvals</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.linalg.eig.html">mlx.core.linalg.eig</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.linalg.eigvalsh.html">mlx.core.linalg.eigvalsh</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.linalg.eigh.html">mlx.core.linalg.eigh</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.linalg.lu.html">mlx.core.linalg.lu</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.linalg.lu_factor.html">mlx.core.linalg.lu_factor</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.linalg.pinv.html">mlx.core.linalg.pinv</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.linalg.solve.html">mlx.core.linalg.solve</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.linalg.solve_triangular.html">mlx.core.linalg.solve_triangular</a></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l1 has-children"><a class="reference internal" href="../python/metal.html">Metal</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.metal.is_available.html">mlx.core.metal.is_available</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.metal.device_info.html">mlx.core.metal.device_info</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.metal.start_capture.html">mlx.core.metal.start_capture</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.metal.stop_capture.html">mlx.core.metal.stop_capture</a></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l1 has-children"><a class="reference internal" href="../python/memory_management.html">Memory Management</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.get_active_memory.html">mlx.core.get_active_memory</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.get_peak_memory.html">mlx.core.get_peak_memory</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.reset_peak_memory.html">mlx.core.reset_peak_memory</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.get_cache_memory.html">mlx.core.get_cache_memory</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.set_memory_limit.html">mlx.core.set_memory_limit</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.set_cache_limit.html">mlx.core.set_cache_limit</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.set_wired_limit.html">mlx.core.set_wired_limit</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.clear_cache.html">mlx.core.clear_cache</a></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l1 has-children"><a class="reference internal" href="../python/nn.html">Neural Networks</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.nn.value_and_grad.html">mlx.nn.value_and_grad</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.nn.quantize.html">mlx.nn.quantize</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.nn.average_gradients.html">mlx.nn.average_gradients</a></li>
|
|
<li class="toctree-l2 has-children"><a class="reference internal" href="../python/nn/module.html">Module</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Module.training.html">mlx.nn.Module.training</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Module.state.html">mlx.nn.Module.state</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Module.apply.html">mlx.nn.Module.apply</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Module.apply_to_modules.html">mlx.nn.Module.apply_to_modules</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Module.children.html">mlx.nn.Module.children</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Module.eval.html">mlx.nn.Module.eval</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Module.filter_and_map.html">mlx.nn.Module.filter_and_map</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Module.freeze.html">mlx.nn.Module.freeze</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Module.leaf_modules.html">mlx.nn.Module.leaf_modules</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Module.load_weights.html">mlx.nn.Module.load_weights</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Module.modules.html">mlx.nn.Module.modules</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Module.named_modules.html">mlx.nn.Module.named_modules</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Module.parameters.html">mlx.nn.Module.parameters</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Module.save_weights.html">mlx.nn.Module.save_weights</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Module.set_dtype.html">mlx.nn.Module.set_dtype</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Module.train.html">mlx.nn.Module.train</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Module.trainable_parameters.html">mlx.nn.Module.trainable_parameters</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Module.unfreeze.html">mlx.nn.Module.unfreeze</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Module.update.html">mlx.nn.Module.update</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Module.update_modules.html">mlx.nn.Module.update_modules</a></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l2 has-children"><a class="reference internal" href="../python/nn/layers.html">Layers</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.ALiBi.html">mlx.nn.ALiBi</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.AvgPool1d.html">mlx.nn.AvgPool1d</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.AvgPool2d.html">mlx.nn.AvgPool2d</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.AvgPool3d.html">mlx.nn.AvgPool3d</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.BatchNorm.html">mlx.nn.BatchNorm</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.CELU.html">mlx.nn.CELU</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Conv1d.html">mlx.nn.Conv1d</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Conv2d.html">mlx.nn.Conv2d</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Conv3d.html">mlx.nn.Conv3d</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.ConvTranspose1d.html">mlx.nn.ConvTranspose1d</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.ConvTranspose2d.html">mlx.nn.ConvTranspose2d</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.ConvTranspose3d.html">mlx.nn.ConvTranspose3d</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Dropout.html">mlx.nn.Dropout</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Dropout2d.html">mlx.nn.Dropout2d</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Dropout3d.html">mlx.nn.Dropout3d</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Embedding.html">mlx.nn.Embedding</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.ELU.html">mlx.nn.ELU</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.GELU.html">mlx.nn.GELU</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.GLU.html">mlx.nn.GLU</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.GroupNorm.html">mlx.nn.GroupNorm</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.GRU.html">mlx.nn.GRU</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.HardShrink.html">mlx.nn.HardShrink</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.HardTanh.html">mlx.nn.HardTanh</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Hardswish.html">mlx.nn.Hardswish</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.InstanceNorm.html">mlx.nn.InstanceNorm</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.LayerNorm.html">mlx.nn.LayerNorm</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.LeakyReLU.html">mlx.nn.LeakyReLU</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Linear.html">mlx.nn.Linear</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.LogSigmoid.html">mlx.nn.LogSigmoid</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.LogSoftmax.html">mlx.nn.LogSoftmax</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.LSTM.html">mlx.nn.LSTM</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.MaxPool1d.html">mlx.nn.MaxPool1d</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.MaxPool2d.html">mlx.nn.MaxPool2d</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.MaxPool3d.html">mlx.nn.MaxPool3d</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Mish.html">mlx.nn.Mish</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.MultiHeadAttention.html">mlx.nn.MultiHeadAttention</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.PReLU.html">mlx.nn.PReLU</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.QuantizedEmbedding.html">mlx.nn.QuantizedEmbedding</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.QuantizedLinear.html">mlx.nn.QuantizedLinear</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.RMSNorm.html">mlx.nn.RMSNorm</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.ReLU.html">mlx.nn.ReLU</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.ReLU6.html">mlx.nn.ReLU6</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.RNN.html">mlx.nn.RNN</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.RoPE.html">mlx.nn.RoPE</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.SELU.html">mlx.nn.SELU</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Sequential.html">mlx.nn.Sequential</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Sigmoid.html">mlx.nn.Sigmoid</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.SiLU.html">mlx.nn.SiLU</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.html">mlx.nn.SinusoidalPositionalEncoding</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Softmin.html">mlx.nn.Softmin</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Softshrink.html">mlx.nn.Softshrink</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Softsign.html">mlx.nn.Softsign</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Softmax.html">mlx.nn.Softmax</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Softplus.html">mlx.nn.Softplus</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Step.html">mlx.nn.Step</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Tanh.html">mlx.nn.Tanh</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Transformer.html">mlx.nn.Transformer</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.Upsample.html">mlx.nn.Upsample</a></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l2 has-children"><a class="reference internal" href="../python/nn/functions.html">Functions</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.elu.html">mlx.nn.elu</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.celu.html">mlx.nn.celu</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.gelu.html">mlx.nn.gelu</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.gelu_approx.html">mlx.nn.gelu_approx</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx.html">mlx.nn.gelu_fast_approx</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.glu.html">mlx.nn.glu</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.hard_shrink.html">mlx.nn.hard_shrink</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.hard_tanh.html">mlx.nn.hard_tanh</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.hardswish.html">mlx.nn.hardswish</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.leaky_relu.html">mlx.nn.leaky_relu</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.log_sigmoid.html">mlx.nn.log_sigmoid</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.log_softmax.html">mlx.nn.log_softmax</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.mish.html">mlx.nn.mish</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.prelu.html">mlx.nn.prelu</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.relu.html">mlx.nn.relu</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.relu6.html">mlx.nn.relu6</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.selu.html">mlx.nn.selu</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.sigmoid.html">mlx.nn.sigmoid</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.silu.html">mlx.nn.silu</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.softmax.html">mlx.nn.softmax</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.softmin.html">mlx.nn.softmin</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.softplus.html">mlx.nn.softplus</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.softshrink.html">mlx.nn.softshrink</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.step.html">mlx.nn.step</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.tanh.html">mlx.nn.tanh</a></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l2 has-children"><a class="reference internal" href="../python/nn/losses.html">Loss Functions</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy.html">mlx.nn.losses.binary_cross_entropy</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss.html">mlx.nn.losses.cosine_similarity_loss</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy.html">mlx.nn.losses.cross_entropy</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.losses.gaussian_nll_loss.html">mlx.nn.losses.gaussian_nll_loss</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss.html">mlx.nn.losses.hinge_loss</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.losses.huber_loss.html">mlx.nn.losses.huber_loss</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.html">mlx.nn.losses.kl_div_loss</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.losses.l1_loss.html">mlx.nn.losses.l1_loss</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss.html">mlx.nn.losses.log_cosh_loss</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.losses.margin_ranking_loss.html">mlx.nn.losses.margin_ranking_loss</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.losses.mse_loss.html">mlx.nn.losses.mse_loss</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.losses.nll_loss.html">mlx.nn.losses.nll_loss</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.html">mlx.nn.losses.smooth_l1_loss</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss.html">mlx.nn.losses.triplet_loss</a></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l2 has-children"><a class="reference internal" href="../python/nn/init.html">Initializers</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.init.constant.html">mlx.nn.init.constant</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.init.normal.html">mlx.nn.init.normal</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.init.uniform.html">mlx.nn.init.uniform</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.init.identity.html">mlx.nn.init.identity</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.init.glorot_normal.html">mlx.nn.init.glorot_normal</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.init.glorot_uniform.html">mlx.nn.init.glorot_uniform</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.init.he_normal.html">mlx.nn.init.he_normal</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/nn/_autosummary/mlx.nn.init.he_uniform.html">mlx.nn.init.he_uniform</a></li>
|
|
</ul>
|
|
</details></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l1 has-children"><a class="reference internal" href="../python/optimizers.html">Optimizers</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l2 has-children"><a class="reference internal" href="../python/optimizers/optimizer.html">Optimizer</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/optimizers/_autosummary/mlx.optimizers.Optimizer.state.html">mlx.optimizers.Optimizer.state</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/optimizers/_autosummary/mlx.optimizers.Optimizer.apply_gradients.html">mlx.optimizers.Optimizer.apply_gradients</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/optimizers/_autosummary/mlx.optimizers.Optimizer.init.html">mlx.optimizers.Optimizer.init</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/optimizers/_autosummary/mlx.optimizers.Optimizer.update.html">mlx.optimizers.Optimizer.update</a></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l2 has-children"><a class="reference internal" href="../python/optimizers/common_optimizers.html">Common Optimizers</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/optimizers/_autosummary/mlx.optimizers.SGD.html">mlx.optimizers.SGD</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/optimizers/_autosummary/mlx.optimizers.RMSprop.html">mlx.optimizers.RMSprop</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/optimizers/_autosummary/mlx.optimizers.Adagrad.html">mlx.optimizers.Adagrad</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/optimizers/_autosummary/mlx.optimizers.Adafactor.html">mlx.optimizers.Adafactor</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/optimizers/_autosummary/mlx.optimizers.AdaDelta.html">mlx.optimizers.AdaDelta</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/optimizers/_autosummary/mlx.optimizers.Adam.html">mlx.optimizers.Adam</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/optimizers/_autosummary/mlx.optimizers.AdamW.html">mlx.optimizers.AdamW</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/optimizers/_autosummary/mlx.optimizers.Adamax.html">mlx.optimizers.Adamax</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/optimizers/_autosummary/mlx.optimizers.Lion.html">mlx.optimizers.Lion</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/optimizers/_autosummary/mlx.optimizers.MultiOptimizer.html">mlx.optimizers.MultiOptimizer</a></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l2 has-children"><a class="reference internal" href="../python/optimizers/schedulers.html">Schedulers</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/optimizers/_autosummary/mlx.optimizers.cosine_decay.html">mlx.optimizers.cosine_decay</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/optimizers/_autosummary/mlx.optimizers.exponential_decay.html">mlx.optimizers.exponential_decay</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/optimizers/_autosummary/mlx.optimizers.join_schedules.html">mlx.optimizers.join_schedules</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/optimizers/_autosummary/mlx.optimizers.linear_schedule.html">mlx.optimizers.linear_schedule</a></li>
|
|
<li class="toctree-l3"><a class="reference internal" href="../python/optimizers/_autosummary/mlx.optimizers.step_decay.html">mlx.optimizers.step_decay</a></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.optimizers.clip_grad_norm.html">mlx.optimizers.clip_grad_norm</a></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l1 has-children"><a class="reference internal" href="../python/distributed.html">Distributed Communication</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.distributed.Group.html">mlx.core.distributed.Group</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.distributed.is_available.html">mlx.core.distributed.is_available</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.distributed.init.html">mlx.core.distributed.init</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.distributed.all_sum.html">mlx.core.distributed.all_sum</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.distributed.all_gather.html">mlx.core.distributed.all_gather</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.distributed.send.html">mlx.core.distributed.send</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.distributed.recv.html">mlx.core.distributed.recv</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.core.distributed.recv_like.html">mlx.core.distributed.recv_like</a></li>
|
|
</ul>
|
|
</details></li>
|
|
<li class="toctree-l1 has-children"><a class="reference internal" href="../python/tree_utils.html">Tree Utils</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.utils.tree_flatten.html">mlx.utils.tree_flatten</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.utils.tree_unflatten.html">mlx.utils.tree_unflatten</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.utils.tree_map.html">mlx.utils.tree_map</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.utils.tree_map_with_path.html">mlx.utils.tree_map_with_path</a></li>
|
|
<li class="toctree-l2"><a class="reference internal" href="../python/_autosummary/mlx.utils.tree_reduce.html">mlx.utils.tree_reduce</a></li>
|
|
</ul>
|
|
</details></li>
|
|
</ul>
|
|
<p aria-level="2" class="caption" role="heading"><span class="caption-text">C++ API Reference</span></p>
|
|
<ul class="current nav bd-sidenav">
|
|
<li class="toctree-l1 current active"><a class="current reference internal" href="#">Operations</a></li>
|
|
</ul>
|
|
<p aria-level="2" class="caption" role="heading"><span class="caption-text">Further Reading</span></p>
|
|
<ul class="nav bd-sidenav">
|
|
<li class="toctree-l1"><a class="reference internal" href="../dev/extensions.html">Custom Extensions in MLX</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../dev/metal_debugger.html">Metal Debugger</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../dev/custom_metal_kernels.html">Custom Metal Kernels</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../dev/mlx_in_cpp.html">Using MLX in C++</a></li>
|
|
</ul>
|
|
|
|
</div>
|
|
</nav></div>
|
|
</div>
|
|
|
|
|
|
<div class="sidebar-primary-items__end sidebar-primary__section">
|
|
</div>
|
|
|
|
<div id="rtd-footer-container"></div>
|
|
|
|
|
|
</div>
|
|
|
|
<main id="main-content" class="bd-main" role="main">
|
|
|
|
|
|
|
|
<div class="sbt-scroll-pixel-helper"></div>
|
|
|
|
<div class="bd-content">
|
|
<div class="bd-article-container">
|
|
|
|
<div class="bd-header-article d-print-none">
|
|
<div class="header-article-items header-article__inner">
|
|
|
|
<div class="header-article-items__start">
|
|
|
|
<div class="header-article-item"><button class="sidebar-toggle primary-toggle btn btn-sm" title="Toggle primary sidebar" data-bs-placement="bottom" data-bs-toggle="tooltip">
|
|
<span class="fa-solid fa-bars"></span>
|
|
</button></div>
|
|
|
|
</div>
|
|
|
|
|
|
<div class="header-article-items__end">
|
|
|
|
<div class="header-article-item">
|
|
|
|
<div class="article-header-buttons">
|
|
|
|
|
|
<a href="https://github.com/ml-explore/mlx" target="_blank"
|
|
class="btn btn-sm btn-source-repository-button"
|
|
title="Source repository"
|
|
data-bs-placement="bottom" data-bs-toggle="tooltip"
|
|
>
|
|
|
|
|
|
<span class="btn__icon-container">
|
|
<i class="fab fa-github"></i>
|
|
</span>
|
|
|
|
</a>
|
|
|
|
|
|
|
|
|
|
|
|
|
|
<div class="dropdown dropdown-download-buttons">
|
|
<button class="btn dropdown-toggle" type="button" data-bs-toggle="dropdown" aria-expanded="false" aria-label="Download this page">
|
|
<i class="fas fa-download"></i>
|
|
</button>
|
|
<ul class="dropdown-menu">
|
|
|
|
|
|
|
|
<li><a href="../_sources/cpp/ops.rst" target="_blank"
|
|
class="btn btn-sm btn-download-source-button dropdown-item"
|
|
title="Download source file"
|
|
data-bs-placement="left" data-bs-toggle="tooltip"
|
|
>
|
|
|
|
|
|
<span class="btn__icon-container">
|
|
<i class="fas fa-file"></i>
|
|
</span>
|
|
<span class="btn__text-container">.rst</span>
|
|
</a>
|
|
</li>
|
|
|
|
|
|
|
|
|
|
<li>
|
|
<button onclick="window.print()"
|
|
class="btn btn-sm btn-download-pdf-button dropdown-item"
|
|
title="Print to PDF"
|
|
data-bs-placement="left" data-bs-toggle="tooltip"
|
|
>
|
|
|
|
|
|
<span class="btn__icon-container">
|
|
<i class="fas fa-file-pdf"></i>
|
|
</span>
|
|
<span class="btn__text-container">.pdf</span>
|
|
</button>
|
|
</li>
|
|
|
|
</ul>
|
|
</div>
|
|
|
|
|
|
|
|
|
|
<button onclick="toggleFullScreen()"
|
|
class="btn btn-sm btn-fullscreen-button"
|
|
title="Fullscreen mode"
|
|
data-bs-placement="bottom" data-bs-toggle="tooltip"
|
|
>
|
|
|
|
|
|
<span class="btn__icon-container">
|
|
<i class="fas fa-expand"></i>
|
|
</span>
|
|
|
|
</button>
|
|
|
|
|
|
|
|
<script>
|
|
document.write(`
|
|
<button class="btn btn-sm nav-link pst-navbar-icon theme-switch-button" title="light/dark" aria-label="light/dark" data-bs-placement="bottom" data-bs-toggle="tooltip">
|
|
<i class="theme-switch fa-solid fa-sun fa-lg" data-mode="light"></i>
|
|
<i class="theme-switch fa-solid fa-moon fa-lg" data-mode="dark"></i>
|
|
<i class="theme-switch fa-solid fa-circle-half-stroke fa-lg" data-mode="auto"></i>
|
|
</button>
|
|
`);
|
|
</script>
|
|
|
|
|
|
<script>
|
|
document.write(`
|
|
<button class="btn btn-sm pst-navbar-icon search-button search-button__button" title="Search" aria-label="Search" data-bs-placement="bottom" data-bs-toggle="tooltip">
|
|
<i class="fa-solid fa-magnifying-glass fa-lg"></i>
|
|
</button>
|
|
`);
|
|
</script>
|
|
<button class="sidebar-toggle secondary-toggle btn btn-sm" title="Toggle secondary sidebar" data-bs-placement="bottom" data-bs-toggle="tooltip">
|
|
<span class="fa-solid fa-list"></span>
|
|
</button>
|
|
</div></div>
|
|
|
|
</div>
|
|
|
|
</div>
|
|
</div>
|
|
|
|
|
|
|
|
<div id="jb-print-docs-body" class="onlyprint">
|
|
<h1>Operations</h1>
|
|
<!-- Table of contents -->
|
|
<div id="print-main-content">
|
|
<div id="jb-print-toc">
|
|
|
|
<div>
|
|
<h2> Contents </h2>
|
|
</div>
|
|
<nav aria-label="Page">
|
|
<ul class="visible nav section-nav flex-column">
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46arangeddd5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arange()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46arangeddd14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arange()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46arangedd5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arange()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46arangedd14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arange()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46aranged5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arange()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46aranged14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arange()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46arangeiii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arange()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46arangeii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arange()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46arangei14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arange()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48linspaceddi5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">linspace()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46astype5array5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">astype()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">as_strided()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44copy5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">copy()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44full5Shape5array5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">full()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44full5Shape5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">full()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">full()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0E4full5array5Shape1T14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">full()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45zerosRK5Shape5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">zeros()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45zerosRK5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">zeros()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410zeros_likeRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">zeros_like()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44onesRK5Shape5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">ones()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44onesRK5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">ones()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49ones_likeRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">ones_like()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43eyeiii5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">eye()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43eyei5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">eye()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43eyeii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">eye()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43eyeiii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">eye()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43eyei14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">eye()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48identityi5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">identity()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48identityi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">identity()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43triiii5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">tri()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43trii5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">tri()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44tril5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">tril()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44triu5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">triu()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47reshapeRK5array5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">reshape()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">unflatten()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47flattenRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">flatten()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47flattenRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">flatten()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">hadamard_transform()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">squeeze()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47squeezeRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">squeeze()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47squeezeRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">squeeze()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">expand_dims()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411expand_dimsRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">expand_dims()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">slice()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">slice()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">slice()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">slice()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">slice_update()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">slice_update()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">slice_update()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45splitRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">split()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45splitRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">split()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">split()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45splitRK5arrayRK5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">split()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">meshgrid()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">clip()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">concatenate()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">concatenate()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">stack()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">stack()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46repeatRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">repeat()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46repeatRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">repeat()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">tile()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">transpose()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">transpose()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48swapaxesRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">swapaxes()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48moveaxisRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">moveaxis()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">pad()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">pad()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">pad()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">pad()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49transposeRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">transpose()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">broadcast_to()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">broadcast_arrays()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45equalRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">equal()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4eqRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator==()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Eeq5array1TRK5array"><code class="docutils literal notranslate"><span class="pre">operator==()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Eeq5arrayRK5array1T"><code class="docutils literal notranslate"><span class="pre">operator==()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49not_equalRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">not_equal()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4neRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator!=()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Ene5array1TRK5array"><code class="docutils literal notranslate"><span class="pre">operator!=()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Ene5arrayRK5array1T"><code class="docutils literal notranslate"><span class="pre">operator!=()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47greaterRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">greater()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4gtRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator>()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Egt5array1TRK5array"><code class="docutils literal notranslate"><span class="pre">operator>()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Egt5arrayRK5array1T"><code class="docutils literal notranslate"><span class="pre">operator>()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">greater_equal()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4geRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator>=()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Ege5array1TRK5array"><code class="docutils literal notranslate"><span class="pre">operator>=()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Ege5arrayRK5array1T"><code class="docutils literal notranslate"><span class="pre">operator>=()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44lessRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">less()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4ltRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator<()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Elt5array1TRK5array"><code class="docutils literal notranslate"><span class="pre">operator<()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Elt5arrayRK5array1T"><code class="docutils literal notranslate"><span class="pre">operator<()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410less_equalRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">less_equal()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4leRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator<=()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Ele5array1TRK5array"><code class="docutils literal notranslate"><span class="pre">operator<=()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Ele5arrayRK5array1T"><code class="docutils literal notranslate"><span class="pre">operator<=()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">array_equal()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411array_equalRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">array_equal()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45isnanRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">isnan()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45isinfRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">isinf()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48isfiniteRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">isfinite()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48isposinfRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">isposinf()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48isneginfRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">isneginf()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">where()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">nan_to_num()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43allRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">all()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43allRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">all()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">allclose()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">isclose()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">all()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43allRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">all()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43anyRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">any()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43anyRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">any()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">any()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43anyRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">any()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43sumRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sum()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43sumRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sum()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sum()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43sumRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sum()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44meanRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">mean()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44meanRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">mean()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">mean()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44meanRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">mean()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43varRK5arraybi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">var()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43varRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">var()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">var()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43varRK5arrayibi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">var()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4StRK5arraybi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">std()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4StRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">std()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">std()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4StRK5arrayibi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">std()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44prodRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">prod()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44prodRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">prod()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">prod()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44prodRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">prod()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43maxRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">max()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43maxRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">max()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">max()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43maxRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">max()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43minRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">min()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43minRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">min()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">min()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43minRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">min()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46argminRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">argmin()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46argminRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">argmin()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46argminRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">argmin()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46argmaxRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">argmax()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46argmaxRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">argmax()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46argmaxRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">argmax()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44sortRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sort()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44sortRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sort()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47argsortRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">argsort()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47argsortRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">argsort()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49partitionRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">partition()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49partitionRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">partition()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412argpartitionRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">argpartition()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412argpartitionRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">argpartition()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44topkRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">topk()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44topkRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">topk()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412logcumsumexpRK5arrayibb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">logcumsumexp()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49logsumexpRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">logsumexp()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49logsumexpRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">logsumexp()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">logsumexp()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49logsumexpRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">logsumexp()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43absRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">abs()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48negativeRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">negative()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4miRK5array"><code class="docutils literal notranslate"><span class="pre">operator-()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44signRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sign()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411logical_notRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">logical_not()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411logical_andRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">logical_and()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4aaRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator&&()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410logical_orRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">logical_or()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4ooRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator||()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410reciprocalRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">reciprocal()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43addRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">add()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4plRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator+()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Epl5array1TRK5array"><code class="docutils literal notranslate"><span class="pre">operator+()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Epl5arrayRK5array1T"><code class="docutils literal notranslate"><span class="pre">operator+()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48subtractRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">subtract()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4miRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator-()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Emi5array1TRK5array"><code class="docutils literal notranslate"><span class="pre">operator-()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Emi5arrayRK5array1T"><code class="docutils literal notranslate"><span class="pre">operator-()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48multiplyRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">multiply()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4mlRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator*()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Eml5array1TRK5array"><code class="docutils literal notranslate"><span class="pre">operator*()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Eml5arrayRK5array1T"><code class="docutils literal notranslate"><span class="pre">operator*()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46divideRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">divide()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4dvRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator/()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4dvdRK5array"><code class="docutils literal notranslate"><span class="pre">operator/()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4dvRK5arrayd"><code class="docutils literal notranslate"><span class="pre">operator/()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46divmodRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">divmod()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">floor_divide()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49remainderRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">remainder()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4rmRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator%()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Erm5array1TRK5array"><code class="docutils literal notranslate"><span class="pre">operator%()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Erm5arrayRK5array1T"><code class="docutils literal notranslate"><span class="pre">operator%()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47maximumRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">maximum()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47minimumRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">minimum()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45floorRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">floor()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44ceilRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">ceil()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46squareRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">square()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43expRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">exp()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43sinRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sin()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43cosRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">cos()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43tanRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">tan()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46arcsinRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arcsin()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46arccosRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arccos()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46arctanRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arctan()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47arctan2RK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arctan2()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44sinhRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sinh()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44coshRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">cosh()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44tanhRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">tanh()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47arcsinhRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arcsinh()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47arccoshRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arccosh()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47arctanhRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arctanh()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47degreesRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">degrees()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47radiansRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">radians()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43logRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">log()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44log2RK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">log2()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45log10RK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">log10()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45log1pRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">log1p()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">logaddexp()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47sigmoidRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sigmoid()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43erfRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">erf()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46erfinvRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">erfinv()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45expm1RK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">expm1()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv413stop_gradientRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">stop_gradient()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45roundRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">round()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45roundRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">round()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46matmulRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">matmul()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">gather()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">gather()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44kronRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">kron()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">take()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44takeRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">take()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44takeRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">take()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44takeRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">take()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">take_along_axis()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">put_along_axis()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter_add_axis()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter_add()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter_add()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter_prod()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter_prod()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter_max()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter_max()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter_min()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter_min()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44sqrtRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sqrt()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45rsqrtRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">rsqrt()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">softmax()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47softmaxRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">softmax()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47softmaxRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">softmax()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45powerRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">power()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46cumsumRK5arrayibb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">cumsum()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47cumprodRK5arrayibb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">cumprod()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46cummaxRK5arrayibb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">cummax()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46cumminRK5arrayibb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">cummin()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">conv_general()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">conv_general()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">conv1d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">conv2d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">conv3d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">conv_transpose1d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">conv_transpose2d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">conv_transpose3d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">quantized_matmul()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48quantizeRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">quantize()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">dequantize()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbiib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">gather_qmm()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">tensordot()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">tensordot()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45outerRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">outer()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45innerRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">inner()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">addmm()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">block_masked_mm()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">gather_mm()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48diagonalRK5arrayiii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">diagonal()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44diagRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">diag()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">trace()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45traceRK5arrayiii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">trace()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45traceRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">trace()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE"><code class="docutils literal notranslate"><span class="pre">depends()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410atleast_1dRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">atleast_1d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">atleast_1d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410atleast_2dRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">atleast_2d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">atleast_2d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410atleast_3dRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">atleast_3d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">atleast_3d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">number_of_elements()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49conjugateRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">conjugate()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">bitwise_and()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4anRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator&()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">bitwise_or()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4orRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator|()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">bitwise_xor()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4eoRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator^()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">left_shift()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4lsRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator<<()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">right_shift()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4rsRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator>>()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv414bitwise_invertRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">bitwise_invert()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4coRK5array"><code class="docutils literal notranslate"><span class="pre">operator~()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">view()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44rollRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">roll()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44rollRK5arrayRK5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">roll()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44rollRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">roll()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">roll()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">roll()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">roll()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44realRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">real()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44imagRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">imag()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410contiguousRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">contiguous()</span></code></a></li>
|
|
</ul>
|
|
</nav>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
|
|
|
|
|
|
<div id="searchbox"></div>
|
|
<article class="bd-article">
|
|
|
|
<section id="operations">
|
|
<span id="cpp-ops"></span><h1>Operations<a class="headerlink" href="#operations" title="Link to this heading">#</a></h1>
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46arangeddd5Dtype14StreamOrDevice">
|
|
<span id="_CPPv36arangeddd5Dtype14StreamOrDevice"></span><span id="_CPPv26arangeddd5Dtype14StreamOrDevice"></span><span id="arange__double.double.double.Dtype.StreamOrDevice"></span><span class="target" id="group__ops_1ga7ca088b8090b9f84f2e08345cf3f835a"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">arange</span></span></span><span class="sig-paren">(</span><span class="kt"><span class="pre">double</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">start</span></span>, <span class="kt"><span class="pre">double</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">stop</span></span>, <span class="kt"><span class="pre">double</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">step</span></span>, <span class="n"><span class="pre">Dtype</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">dtype</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46arangeddd5Dtype14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>A 1D array of numbers starting at <code class="docutils literal notranslate"><span class="pre">start</span></code> (optional), stopping at stop, stepping by <code class="docutils literal notranslate"><span class="pre">step</span></code> (optional). </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46arangeddd14StreamOrDevice">
|
|
<span id="_CPPv36arangeddd14StreamOrDevice"></span><span id="_CPPv26arangeddd14StreamOrDevice"></span><span id="arange__double.double.double.StreamOrDevice"></span><span class="target" id="group__ops_1ga4c36b841dc5cba391dad029be5a0ad98"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">arange</span></span></span><span class="sig-paren">(</span><span class="kt"><span class="pre">double</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">start</span></span>, <span class="kt"><span class="pre">double</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">stop</span></span>, <span class="kt"><span class="pre">double</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">step</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46arangeddd14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46arangedd5Dtype14StreamOrDevice">
|
|
<span id="_CPPv36arangedd5Dtype14StreamOrDevice"></span><span id="_CPPv26arangedd5Dtype14StreamOrDevice"></span><span id="arange__double.double.Dtype.StreamOrDevice"></span><span class="target" id="group__ops_1ga8d7cf9eb15e2daf1469058907e8abc85"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">arange</span></span></span><span class="sig-paren">(</span><span class="kt"><span class="pre">double</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">start</span></span>, <span class="kt"><span class="pre">double</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">stop</span></span>, <span class="n"><span class="pre">Dtype</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">dtype</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46arangedd5Dtype14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46arangedd14StreamOrDevice">
|
|
<span id="_CPPv36arangedd14StreamOrDevice"></span><span id="_CPPv26arangedd14StreamOrDevice"></span><span id="arange__double.double.StreamOrDevice"></span><span class="target" id="group__ops_1ga74566a14e69ba6a25f5a35e7ade5c282"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">arange</span></span></span><span class="sig-paren">(</span><span class="kt"><span class="pre">double</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">start</span></span>, <span class="kt"><span class="pre">double</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">stop</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46arangedd14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46aranged5Dtype14StreamOrDevice">
|
|
<span id="_CPPv36aranged5Dtype14StreamOrDevice"></span><span id="_CPPv26aranged5Dtype14StreamOrDevice"></span><span id="arange__double.Dtype.StreamOrDevice"></span><span class="target" id="group__ops_1ga345aa27af3dae3646b8b4b1068e89a3e"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">arange</span></span></span><span class="sig-paren">(</span><span class="kt"><span class="pre">double</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">stop</span></span>, <span class="n"><span class="pre">Dtype</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">dtype</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46aranged5Dtype14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46aranged14StreamOrDevice">
|
|
<span id="_CPPv36aranged14StreamOrDevice"></span><span id="_CPPv26aranged14StreamOrDevice"></span><span id="arange__double.StreamOrDevice"></span><span class="target" id="group__ops_1gaae179075d0fe23f4bd53fdf8c41f4c70"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">arange</span></span></span><span class="sig-paren">(</span><span class="kt"><span class="pre">double</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">stop</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46aranged14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46arangeiii14StreamOrDevice">
|
|
<span id="_CPPv36arangeiii14StreamOrDevice"></span><span id="_CPPv26arangeiii14StreamOrDevice"></span><span id="arange__i.i.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga6b945f513077c2978afc1a952c884860"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">arange</span></span></span><span class="sig-paren">(</span><span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">start</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">stop</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">step</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46arangeiii14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46arangeii14StreamOrDevice">
|
|
<span id="_CPPv36arangeii14StreamOrDevice"></span><span id="_CPPv26arangeii14StreamOrDevice"></span><span id="arange__i.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga1c39fcc6eaa1c1867735c7f849d708d6"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">arange</span></span></span><span class="sig-paren">(</span><span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">start</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">stop</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46arangeii14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46arangei14StreamOrDevice">
|
|
<span id="_CPPv36arangei14StreamOrDevice"></span><span id="_CPPv26arangei14StreamOrDevice"></span><span id="arange__i.StreamOrDevice"></span><span class="target" id="group__ops_1gafe6e4580452c873cac294f16129e633f"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">arange</span></span></span><span class="sig-paren">(</span><span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">stop</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46arangei14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv48linspaceddi5Dtype14StreamOrDevice">
|
|
<span id="_CPPv38linspaceddi5Dtype14StreamOrDevice"></span><span id="_CPPv28linspaceddi5Dtype14StreamOrDevice"></span><span id="linspace__double.double.i.Dtype.StreamOrDevice"></span><span class="target" id="group__ops_1ga968bcabed902311dcfbd903b0fb886ec"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">linspace</span></span></span><span class="sig-paren">(</span><span class="kt"><span class="pre">double</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">start</span></span>, <span class="kt"><span class="pre">double</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">stop</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">num</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">50</span></span>, <span class="n"><span class="pre">Dtype</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">dtype</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="n"><span class="pre">float32</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv48linspaceddi5Dtype14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>A 1D array of <code class="docutils literal notranslate"><span class="pre">num</span></code> evenly spaced numbers in the range <code class="docutils literal notranslate"><span class="pre">[start,</span> <span class="pre">stop]</span></code></p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46astype5array5Dtype14StreamOrDevice">
|
|
<span id="_CPPv36astype5array5Dtype14StreamOrDevice"></span><span id="_CPPv26astype5array5Dtype14StreamOrDevice"></span><span id="astype__array.Dtype.StreamOrDevice"></span><span class="target" id="group__ops_1ga0e58c24fc5668e5a521e5b45e8370a62"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">astype</span></span></span><span class="sig-paren">(</span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">Dtype</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">dtype</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46astype5array5Dtype14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Convert an array to the given data type. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice">
|
|
<span id="_CPPv310as_strided5array5Shape7Strides6size_t14StreamOrDevice"></span><span id="_CPPv210as_strided5array5Shape7Strides6size_t14StreamOrDevice"></span><span id="as_strided__array.Shape.Strides.s.StreamOrDevice"></span><span class="target" id="group__ops_1ga6085b03f2662ef2a61de523fd609f3bf"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">as_strided</span></span></span><span class="sig-paren">(</span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">shape</span></span>, <span class="n"><span class="pre">Strides</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">strides</span></span>, <span class="n"><span class="pre">size_t</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">offset</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Create a view of an array with the given shape and strides. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44copy5array14StreamOrDevice">
|
|
<span id="_CPPv34copy5array14StreamOrDevice"></span><span id="_CPPv24copy5array14StreamOrDevice"></span><span id="copy__array.StreamOrDevice"></span><span class="target" id="group__ops_1gae306e93af12f774bd80bad6c231b09d6"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">copy</span></span></span><span class="sig-paren">(</span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44copy5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Copy another array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44full5Shape5array5Dtype14StreamOrDevice">
|
|
<span id="_CPPv34full5Shape5array5Dtype14StreamOrDevice"></span><span id="_CPPv24full5Shape5array5Dtype14StreamOrDevice"></span><span id="full__Shape.array.Dtype.StreamOrDevice"></span><span class="target" id="group__ops_1ga1cf232308668fe3f4214c8b895ed4aee"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">full</span></span></span><span class="sig-paren">(</span><span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">shape</span></span>, <span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">vals</span></span>, <span class="n"><span class="pre">Dtype</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">dtype</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44full5Shape5array5Dtype14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Fill an array of the given shape with the given value(s). </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44full5Shape5array14StreamOrDevice">
|
|
<span id="_CPPv34full5Shape5array14StreamOrDevice"></span><span id="_CPPv24full5Shape5array14StreamOrDevice"></span><span id="full__Shape.array.StreamOrDevice"></span><span class="target" id="group__ops_1ga59f6c844cbb173e108c3eeb11801f8c6"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">full</span></span></span><span class="sig-paren">(</span><span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">shape</span></span>, <span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">vals</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44full5Shape5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice">
|
|
<span id="_CPPv3I0E4full5Shape1T5Dtype14StreamOrDevice"></span><span id="_CPPv2I0E4full5Shape1T5Dtype14StreamOrDevice"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1gaf073760b7b51fe35932da0d81c531a55"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">full</span></span></span><span class="sig-paren">(</span><span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">shape</span></span>, <a class="reference internal" href="#_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice" title="full::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">val</span></span>, <span class="n"><span class="pre">Dtype</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">dtype</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0E4full5array5Shape1T14StreamOrDevice">
|
|
<span id="_CPPv3I0E4full5Shape1T14StreamOrDevice"></span><span id="_CPPv2I0E4full5Shape1T14StreamOrDevice"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1gaf6f2cce92aff9b71756a3cc3c961fd5a"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">full</span></span></span><span class="sig-paren">(</span><span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">shape</span></span>, <a class="reference internal" href="#_CPPv4I0E4full5array5Shape1T14StreamOrDevice" title="full::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">val</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0E4full5array5Shape1T14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45zerosRK5Shape5Dtype14StreamOrDevice">
|
|
<span id="_CPPv35zerosRK5Shape5Dtype14StreamOrDevice"></span><span id="_CPPv25zerosRK5Shape5Dtype14StreamOrDevice"></span><span id="zeros__ShapeCR.Dtype.StreamOrDevice"></span><span class="target" id="group__ops_1gae2cace3b388cec4e520659a91879e1c1"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">zeros</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">shape</span></span>, <span class="n"><span class="pre">Dtype</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">dtype</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45zerosRK5Shape5Dtype14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Fill an array of the given shape with zeros. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45zerosRK5Shape14StreamOrDevice">
|
|
<span id="_CPPv35zerosRK5Shape14StreamOrDevice"></span><span id="_CPPv25zerosRK5Shape14StreamOrDevice"></span><span id="zeros__ShapeCR.StreamOrDevice"></span><span class="target" id="group__ops_1gac8aa722f5e798819b7091693173f1f36"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">zeros</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">shape</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45zerosRK5Shape14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv410zeros_likeRK5array14StreamOrDevice">
|
|
<span id="_CPPv310zeros_likeRK5array14StreamOrDevice"></span><span id="_CPPv210zeros_likeRK5array14StreamOrDevice"></span><span id="zeros_like__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gafbb857094d784b38c78683a091ffdbde"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">zeros_like</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv410zeros_likeRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44onesRK5Shape5Dtype14StreamOrDevice">
|
|
<span id="_CPPv34onesRK5Shape5Dtype14StreamOrDevice"></span><span id="_CPPv24onesRK5Shape5Dtype14StreamOrDevice"></span><span id="ones__ShapeCR.Dtype.StreamOrDevice"></span><span class="target" id="group__ops_1gae0069146cf8c819b15ba29aa7231a3f0"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">ones</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">shape</span></span>, <span class="n"><span class="pre">Dtype</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">dtype</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44onesRK5Shape5Dtype14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Fill an array of the given shape with ones. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44onesRK5Shape14StreamOrDevice">
|
|
<span id="_CPPv34onesRK5Shape14StreamOrDevice"></span><span id="_CPPv24onesRK5Shape14StreamOrDevice"></span><span id="ones__ShapeCR.StreamOrDevice"></span><span class="target" id="group__ops_1gace4cf016562af58991f9f961170e156f"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">ones</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">shape</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44onesRK5Shape14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv49ones_likeRK5array14StreamOrDevice">
|
|
<span id="_CPPv39ones_likeRK5array14StreamOrDevice"></span><span id="_CPPv29ones_likeRK5array14StreamOrDevice"></span><span id="ones_like__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga94f8d3b1906fee99da9cbe39f7be7d42"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">ones_like</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv49ones_likeRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43eyeiii5Dtype14StreamOrDevice">
|
|
<span id="_CPPv33eyeiii5Dtype14StreamOrDevice"></span><span id="_CPPv23eyeiii5Dtype14StreamOrDevice"></span><span id="eye__i.i.i.Dtype.StreamOrDevice"></span><span class="target" id="group__ops_1ga45e9e68246b0d1cf03c3cc9c9e7e6ae3"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">eye</span></span></span><span class="sig-paren">(</span><span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">n</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">m</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">k</span></span>, <span class="n"><span class="pre">Dtype</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">dtype</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43eyeiii5Dtype14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Fill an array of the given shape (n,m) with ones in the specified diagonal k, and zeros everywhere else. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43eyei5Dtype14StreamOrDevice">
|
|
<span id="_CPPv33eyei5Dtype14StreamOrDevice"></span><span id="_CPPv23eyei5Dtype14StreamOrDevice"></span><span id="eye__i.Dtype.StreamOrDevice"></span><span class="target" id="group__ops_1ga2c9011310a1fa7c82f942f54102c36dd"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">eye</span></span></span><span class="sig-paren">(</span><span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">n</span></span>, <span class="n"><span class="pre">Dtype</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">dtype</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43eyei5Dtype14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43eyeii14StreamOrDevice">
|
|
<span id="_CPPv33eyeii14StreamOrDevice"></span><span id="_CPPv23eyeii14StreamOrDevice"></span><span id="eye__i.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga61657db78ef35d41112d362c869c25d2"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">eye</span></span></span><span class="sig-paren">(</span><span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">n</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">m</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43eyeii14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43eyeiii14StreamOrDevice">
|
|
<span id="_CPPv33eyeiii14StreamOrDevice"></span><span id="_CPPv23eyeiii14StreamOrDevice"></span><span id="eye__i.i.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga908a15b42834be498a46856c99dfc779"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">eye</span></span></span><span class="sig-paren">(</span><span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">n</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">m</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">k</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43eyeiii14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43eyei14StreamOrDevice">
|
|
<span id="_CPPv33eyei14StreamOrDevice"></span><span id="_CPPv23eyei14StreamOrDevice"></span><span id="eye__i.StreamOrDevice"></span><span class="target" id="group__ops_1gab777fcf6d4a89172c69ec3492548dc0f"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">eye</span></span></span><span class="sig-paren">(</span><span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">n</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43eyei14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv48identityi5Dtype14StreamOrDevice">
|
|
<span id="_CPPv38identityi5Dtype14StreamOrDevice"></span><span id="_CPPv28identityi5Dtype14StreamOrDevice"></span><span id="identity__i.Dtype.StreamOrDevice"></span><span class="target" id="group__ops_1ga484eaa10d5e19a4ca46d3a9cd9fab600"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">identity</span></span></span><span class="sig-paren">(</span><span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">n</span></span>, <span class="n"><span class="pre">Dtype</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">dtype</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv48identityi5Dtype14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Create a square matrix of shape (n,n) of zeros, and ones in the major diagonal. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv48identityi14StreamOrDevice">
|
|
<span id="_CPPv38identityi14StreamOrDevice"></span><span id="_CPPv28identityi14StreamOrDevice"></span><span id="identity__i.StreamOrDevice"></span><span class="target" id="group__ops_1gad994d65ac6019c26b5ad6c41179d3424"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">identity</span></span></span><span class="sig-paren">(</span><span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">n</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv48identityi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43triiii5Dtype14StreamOrDevice">
|
|
<span id="_CPPv33triiii5Dtype14StreamOrDevice"></span><span id="_CPPv23triiii5Dtype14StreamOrDevice"></span><span id="tri__i.i.i.Dtype.StreamOrDevice"></span><span class="target" id="group__ops_1ga4f3389e5b89e70e862e7d2b40d6c7f78"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">tri</span></span></span><span class="sig-paren">(</span><span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">n</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">m</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">k</span></span>, <span class="n"><span class="pre">Dtype</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">type</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43triiii5Dtype14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43trii5Dtype14StreamOrDevice">
|
|
<span id="_CPPv33trii5Dtype14StreamOrDevice"></span><span id="_CPPv23trii5Dtype14StreamOrDevice"></span><span id="tri__i.Dtype.StreamOrDevice"></span><span class="target" id="group__ops_1gac19a1bd6ed6d5c7bc9d258820189dbb5"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">tri</span></span></span><span class="sig-paren">(</span><span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">n</span></span>, <span class="n"><span class="pre">Dtype</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">type</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43trii5Dtype14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44tril5arrayi14StreamOrDevice">
|
|
<span id="_CPPv34tril5arrayi14StreamOrDevice"></span><span id="_CPPv24tril5arrayi14StreamOrDevice"></span><span id="tril__array.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga83e0bb45dc770cf014531d873b78c5a2"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">tril</span></span></span><span class="sig-paren">(</span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">x</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">k</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44tril5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44triu5arrayi14StreamOrDevice">
|
|
<span id="_CPPv34triu5arrayi14StreamOrDevice"></span><span id="_CPPv24triu5arrayi14StreamOrDevice"></span><span id="triu__array.i.StreamOrDevice"></span><span class="target" id="group__ops_1gaa9df5917876eeb0cb28b7fa81f880412"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">triu</span></span></span><span class="sig-paren">(</span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">x</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">k</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44triu5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47reshapeRK5array5Shape14StreamOrDevice">
|
|
<span id="_CPPv37reshapeRK5array5Shape14StreamOrDevice"></span><span id="_CPPv27reshapeRK5array5Shape14StreamOrDevice"></span><span id="reshape__arrayCR.Shape.StreamOrDevice"></span><span class="target" id="group__ops_1ga084f03ce2b22258afb7c8b45e17af828"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">reshape</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">shape</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47reshapeRK5array5Shape14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Reshape an array to the given shape. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice">
|
|
<span id="_CPPv39unflattenRK5arrayi5Shape14StreamOrDevice"></span><span id="_CPPv29unflattenRK5arrayi5Shape14StreamOrDevice"></span><span id="unflatten__arrayCR.i.Shape.StreamOrDevice"></span><span class="target" id="group__ops_1ga666bcc2187a144247e8c0c224b016625"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">unflatten</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">shape</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Unflatten the axis to the given shape. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47flattenRK5arrayii14StreamOrDevice">
|
|
<span id="_CPPv37flattenRK5arrayii14StreamOrDevice"></span><span id="_CPPv27flattenRK5arrayii14StreamOrDevice"></span><span id="flatten__arrayCR.i.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga50aa98754b412bb57c083f6e3e95061f"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">flatten</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">start_axis</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">end_axis</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="o"><span class="pre">-</span></span><span class="m"><span class="pre">1</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47flattenRK5arrayii14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Flatten the dimensions in the range <code class="docutils literal notranslate"><span class="pre">[start_axis,</span> <span class="pre">end_axis]</span></code> . </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47flattenRK5array14StreamOrDevice">
|
|
<span id="_CPPv37flattenRK5array14StreamOrDevice"></span><span id="_CPPv27flattenRK5array14StreamOrDevice"></span><span id="flatten__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gaa6adbc9c86f0ab27d8810a02e9e719fd"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">flatten</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47flattenRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Flatten the array to 1D. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice">
|
|
<span id="_CPPv318hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice"></span><span id="_CPPv218hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice"></span><span id="hadamard_transform__arrayCR.std::optional:float:.StreamOrDevice"></span><span class="target" id="group__ops_1ga872d2c1806e67ce2596b24d056681074"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">hadamard_transform</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">optional</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">float</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">scale</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">nullopt</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Multiply the array by the Hadamard matrix of corresponding size. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice">
|
|
<span id="_CPPv37squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice"></span><span id="_CPPv27squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice"></span><span id="squeeze__arrayCR.std::vector:i:CR.StreamOrDevice"></span><span class="target" id="group__ops_1ga710daa7ec721bd4d3f326082cb195576"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">squeeze</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Remove singleton dimensions at the given axes. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47squeezeRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv37squeezeRK5arrayi14StreamOrDevice"></span><span id="_CPPv27squeezeRK5arrayi14StreamOrDevice"></span><span id="squeeze__arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga700dd51b77379a3d2260a55783e8ebf3"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">squeeze</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47squeezeRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Remove singleton dimensions at the given axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47squeezeRK5array14StreamOrDevice">
|
|
<span id="_CPPv37squeezeRK5array14StreamOrDevice"></span><span id="_CPPv27squeezeRK5array14StreamOrDevice"></span><span id="squeeze__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga58bad3c61fd85b95927a987ba1cf5dad"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">squeeze</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47squeezeRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Remove all singleton dimensions. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice">
|
|
<span id="_CPPv311expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice"></span><span id="_CPPv211expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice"></span><span id="expand_dims__arrayCR.std::vector:i:CR.StreamOrDevice"></span><span class="target" id="group__ops_1ga717f11149a8c7b4cc3e33bbcc0a97133"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">expand_dims</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Add a singleton dimension at the given axes. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv411expand_dimsRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv311expand_dimsRK5arrayi14StreamOrDevice"></span><span id="_CPPv211expand_dimsRK5arrayi14StreamOrDevice"></span><span id="expand_dims__arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga7a80adb4a5a36d18b5f234d4b034950a"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">expand_dims</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv411expand_dimsRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Add a singleton dimension at the given axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice">
|
|
<span id="_CPPv35sliceRK5array5Shape5Shape5Shape14StreamOrDevice"></span><span id="_CPPv25sliceRK5array5Shape5Shape5Shape14StreamOrDevice"></span><span id="slice__arrayCR.Shape.Shape.Shape.StreamOrDevice"></span><span class="target" id="group__ops_1ga29718cd5005dbcde0396b6fd65cc041d"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">slice</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">start</span></span>, <span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">stop</span></span>, <span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">strides</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Slice an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice">
|
|
<span id="_CPPv35sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice"></span><span id="_CPPv25sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice"></span><span id="slice__arrayCR.std::initializer_list:i:.Shape.Shape.StreamOrDevice"></span><span class="target" id="group__ops_1gaedcbdf4040f5e6a02a74b1ed1c6c2ebc"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">slice</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">initializer_list</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">start</span></span>, <span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">stop</span></span>, <span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">strides</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice">
|
|
<span id="_CPPv35sliceRK5array5Shape5Shape14StreamOrDevice"></span><span id="_CPPv25sliceRK5array5Shape5Shape14StreamOrDevice"></span><span id="slice__arrayCR.Shape.Shape.StreamOrDevice"></span><span class="target" id="group__ops_1gaec56dcb94d5e7f7b885fb60b4bf4aa9d"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">slice</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">start</span></span>, <span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">stop</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Slice an array with a stride of 1 in each dimension. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice">
|
|
<span id="_CPPv35sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice"></span><span id="_CPPv25sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice"></span><span id="slice__arrayCR.arrayCR.std::vector:i:.Shape.StreamOrDevice"></span><span class="target" id="group__ops_1ga797996e53ea34317a55dc2f314736b89"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">slice</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">start</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">slice_size</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Slice an array with dynamic starting indices. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice">
|
|
<span id="_CPPv312slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice"></span><span id="_CPPv212slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice"></span><span id="slice_update__arrayCR.arrayCR.Shape.Shape.Shape.StreamOrDevice"></span><span class="target" id="group__ops_1ga7b7e786985d27789aaed20bb2f9509be"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">slice_update</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">src</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">update</span></span>, <span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">start</span></span>, <span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">stop</span></span>, <span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">strides</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Update a slice from the source array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice">
|
|
<span id="_CPPv312slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice"></span><span id="_CPPv212slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice"></span><span id="slice_update__arrayCR.arrayCR.Shape.Shape.StreamOrDevice"></span><span class="target" id="group__ops_1ga3b7c6136ae5a38dd6457b65c1833aa67"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">slice_update</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">src</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">update</span></span>, <span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">start</span></span>, <span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">stop</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Update a slice from the source array with stride 1 in each dimension. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice">
|
|
<span id="_CPPv312slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice"></span><span id="_CPPv212slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice"></span><span id="slice_update__arrayCR.arrayCR.arrayCR.std::vector:i:.StreamOrDevice"></span><span class="target" id="group__ops_1gacd906ffb96149a4998c321cbf2231d7e"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">slice_update</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">src</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">update</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">start</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Update a slice from the source array with dynamic starting indices. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45splitRK5arrayii14StreamOrDevice">
|
|
<span id="_CPPv35splitRK5arrayii14StreamOrDevice"></span><span id="_CPPv25splitRK5arrayii14StreamOrDevice"></span><span id="split__arrayCR.i.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga7534290bceab5fb3831a05d67bebce7d"></span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">split</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">num_splits</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45splitRK5arrayii14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Split an array into sub-arrays along a given axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45splitRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv35splitRK5arrayi14StreamOrDevice"></span><span id="_CPPv25splitRK5arrayi14StreamOrDevice"></span><span id="split__arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga56882d24e5fde59c266774624c892d41"></span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">split</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">num_splits</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45splitRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice">
|
|
<span id="_CPPv35splitRK5arrayRK5Shapei14StreamOrDevice"></span><span id="_CPPv25splitRK5arrayRK5Shapei14StreamOrDevice"></span><span id="split__arrayCR.ShapeCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga19005414e7d8020cd6e94e06bf399b09"></span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">split</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">indices</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45splitRK5arrayRK5Shape14StreamOrDevice">
|
|
<span id="_CPPv35splitRK5arrayRK5Shape14StreamOrDevice"></span><span id="_CPPv25splitRK5arrayRK5Shape14StreamOrDevice"></span><span id="split__arrayCR.ShapeCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga9ea089f42b9940510619052b7166d9ac"></span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">split</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">indices</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45splitRK5arrayRK5Shape14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice">
|
|
<span id="_CPPv38meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice"></span><span id="_CPPv28meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice"></span><span id="meshgrid__std::vector:array:CR.b.ssCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga5ecddb74ba7861eb82eca8653501d5dc"></span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">meshgrid</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">arrays</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">sparse</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">string</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">indexing</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="s"><span class="pre">"xy"</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>A vector of coordinate arrays from coordinate vectors. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice">
|
|
<span id="_CPPv34clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice"></span><span id="_CPPv24clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice"></span><span id="clip__arrayCR.std::optional:array:CR.std::optional:array:CR.StreamOrDevice"></span><span class="target" id="group__ops_1ga157cd7c23f9b306fee2e1eb2b9bf1dd8"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">clip</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">optional</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a_min</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">nullopt</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">optional</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a_max</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">nullopt</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Clip (limit) the values in an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice">
|
|
<span id="_CPPv311concatenateNSt6vectorI5arrayEEi14StreamOrDevice"></span><span id="_CPPv211concatenateNSt6vectorI5arrayEEi14StreamOrDevice"></span><span id="concatenate__std::vector:array:.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga52838af566948b1b96e7aa00832071b3"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">concatenate</span></span></span><span class="sig-paren">(</span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">arrays</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Concatenate arrays along a given axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice">
|
|
<span id="_CPPv311concatenateNSt6vectorI5arrayEE14StreamOrDevice"></span><span id="_CPPv211concatenateNSt6vectorI5arrayEE14StreamOrDevice"></span><span id="concatenate__std::vector:array:.StreamOrDevice"></span><span class="target" id="group__ops_1ga666ac69778984fafdc2f51d296270468"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">concatenate</span></span></span><span class="sig-paren">(</span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">arrays</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice">
|
|
<span id="_CPPv35stackRKNSt6vectorI5arrayEEi14StreamOrDevice"></span><span id="_CPPv25stackRKNSt6vectorI5arrayEEi14StreamOrDevice"></span><span id="stack__std::vector:array:CR.i.StreamOrDevice"></span><span class="target" id="group__ops_1gaf8f2ec2b98a4b59eca73d7471df6e032"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">stack</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">arrays</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Stack arrays along a new axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice">
|
|
<span id="_CPPv35stackRKNSt6vectorI5arrayEE14StreamOrDevice"></span><span id="_CPPv25stackRKNSt6vectorI5arrayEE14StreamOrDevice"></span><span id="stack__std::vector:array:CR.StreamOrDevice"></span><span class="target" id="group__ops_1ga82216209dce901296fc737fe8efa5c94"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">stack</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">arrays</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46repeatRK5arrayii14StreamOrDevice">
|
|
<span id="_CPPv36repeatRK5arrayii14StreamOrDevice"></span><span id="_CPPv26repeatRK5arrayii14StreamOrDevice"></span><span id="repeat__arrayCR.i.i.StreamOrDevice"></span><span class="target" id="group__ops_1gab49e3a687e826554ed1574186e8ae974"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">repeat</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">arr</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">repeats</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46repeatRK5arrayii14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Repeat an array along an axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46repeatRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv36repeatRK5arrayi14StreamOrDevice"></span><span id="_CPPv26repeatRK5arrayi14StreamOrDevice"></span><span id="repeat__arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga4f75f5d5db999f02f43ecbc6dccf3ba6"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">repeat</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">arr</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">repeats</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46repeatRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice">
|
|
<span id="_CPPv34tileRK5arrayNSt6vectorIiEE14StreamOrDevice"></span><span id="_CPPv24tileRK5arrayNSt6vectorIiEE14StreamOrDevice"></span><span id="tile__arrayCR.std::vector:i:.StreamOrDevice"></span><span class="target" id="group__ops_1gab105a57b9a4d84496fe1e4d60e13d361"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">tile</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">arr</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">reps</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice">
|
|
<span id="_CPPv39transposeRK5arrayNSt6vectorIiEE14StreamOrDevice"></span><span id="_CPPv29transposeRK5arrayNSt6vectorIiEE14StreamOrDevice"></span><span id="transpose__arrayCR.std::vector:i:.StreamOrDevice"></span><span class="target" id="group__ops_1gac1869f3b7094869b44fe7ac4ce58638b"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">transpose</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Permutes the dimensions according to the given axes. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice">
|
|
<span id="_CPPv39transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice"></span><span id="_CPPv29transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice"></span><span id="transpose__arrayCR.std::initializer_list:i:.StreamOrDevice"></span><span class="target" id="group__ops_1ga260ac332956f3a6bf1dfdb9095c84dc5"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">transpose</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">initializer_list</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv48swapaxesRK5arrayii14StreamOrDevice">
|
|
<span id="_CPPv38swapaxesRK5arrayii14StreamOrDevice"></span><span id="_CPPv28swapaxesRK5arrayii14StreamOrDevice"></span><span id="swapaxes__arrayCR.i.i.StreamOrDevice"></span><span class="target" id="group__ops_1gabc46eed81ab6c6247903e4ec0c4ec1fb"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">swapaxes</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis1</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis2</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv48swapaxesRK5arrayii14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Swap two axes of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv48moveaxisRK5arrayii14StreamOrDevice">
|
|
<span id="_CPPv38moveaxisRK5arrayii14StreamOrDevice"></span><span id="_CPPv28moveaxisRK5arrayii14StreamOrDevice"></span><span id="moveaxis__arrayCR.i.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga24067d10a842db2c9d509ea48135a2c3"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">moveaxis</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">source</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">destination</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv48moveaxisRK5arrayii14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Move an axis of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice">
|
|
<span id="_CPPv33padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice"></span><span id="_CPPv23padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice"></span><span id="pad__arrayCR.std::vector:i:CR.ShapeCR.ShapeCR.arrayCR.ssCR.StreamOrDevice"></span><span class="target" id="group__ops_1gab95ebd20bd7c6d1c840007cc020cbc0c"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">pad</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">low_pad_size</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">high_pad_size</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">pad_value</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">(</span></span><span class="m"><span class="pre">0</span></span><span class="p"><span class="pre">)</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">string</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">mode</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="s"><span class="pre">"constant"</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Pad an array with a constant value. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice">
|
|
<span id="_CPPv33padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice"></span><span id="_CPPv23padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice"></span><span id="pad__arrayCR.std::vector:std::pair:i.i::CR.arrayCR.ssCR.StreamOrDevice"></span><span class="target" id="group__ops_1gad89f464e92c356faab1f1c2e763b1fb7"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">pad</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">pair</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">pad_width</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">pad_value</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">(</span></span><span class="m"><span class="pre">0</span></span><span class="p"><span class="pre">)</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">string</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">mode</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="s"><span class="pre">"constant"</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Pad an array with a constant value along all axes. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice">
|
|
<span id="_CPPv33padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice"></span><span id="_CPPv23padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice"></span><span id="pad__arrayCR.std::pair:i.i:CR.arrayCR.ssCR.StreamOrDevice"></span><span class="target" id="group__ops_1gab2d5a17d3d5225fed34905e786c31c5f"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">pad</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">pair</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">pad_width</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">pad_value</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">(</span></span><span class="m"><span class="pre">0</span></span><span class="p"><span class="pre">)</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">string</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">mode</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="s"><span class="pre">"constant"</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice">
|
|
<span id="_CPPv33padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice"></span><span id="_CPPv23padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice"></span><span id="pad__arrayCR.i.arrayCR.ssCR.StreamOrDevice"></span><span class="target" id="group__ops_1gaa73ac5674467b1d5c74de6fef7204c44"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">pad</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">pad_width</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">pad_value</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">(</span></span><span class="m"><span class="pre">0</span></span><span class="p"><span class="pre">)</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">string</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">mode</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="s"><span class="pre">"constant"</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv49transposeRK5array14StreamOrDevice">
|
|
<span id="_CPPv39transposeRK5array14StreamOrDevice"></span><span id="_CPPv29transposeRK5array14StreamOrDevice"></span><span id="transpose__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga68da0176fefbe0c0096783c6fd926c6a"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">transpose</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv49transposeRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Permutes the dimensions in reverse order. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice">
|
|
<span id="_CPPv312broadcast_toRK5arrayRK5Shape14StreamOrDevice"></span><span id="_CPPv212broadcast_toRK5arrayRK5Shape14StreamOrDevice"></span><span id="broadcast_to__arrayCR.ShapeCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga2fd5891f11593b7f09550e884f969013"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">broadcast_to</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">shape</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Broadcast an array to a given shape. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice">
|
|
<span id="_CPPv316broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice"></span><span id="_CPPv216broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice"></span><span id="broadcast_arrays__std::vector:array:CR.StreamOrDevice"></span><span class="target" id="group__ops_1gab783890428b596f715dc7dd2057eae99"></span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">broadcast_arrays</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">inputs</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Broadcast a vector of arrays against one another. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45equalRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv35equalRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv25equalRK5arrayRK5array14StreamOrDevice"></span><span id="equal__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga33638dc3a9972dd02be12d0eb85f9bde"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">equal</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45equalRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns the bool array with (a == b) element-wise. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4eqRK5arrayRK5array">
|
|
<span id="_CPPv3eqRK5arrayRK5array"></span><span id="_CPPv2eqRK5arrayRK5array"></span><span id="eq-operator__arrayCR.arrayCR"></span><span class="target" id="group__ops_1gaa30cf69f3d22f65615f5e1696dd5703f"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">==</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4eqRK5arrayRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0Eeq5array1TRK5array">
|
|
<span id="_CPPv3I0Eeq1TRK5array"></span><span id="_CPPv2I0Eeq1TRK5array"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1gaf115782d009ac2a547fcca395c9ec797"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">==</span></span></span><span class="sig-paren">(</span><a class="reference internal" href="#_CPPv4I0Eeq5array1TRK5array" title="operator==::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0Eeq5array1TRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0Eeq5arrayRK5array1T">
|
|
<span id="_CPPv3I0EeqRK5array1T"></span><span id="_CPPv2I0EeqRK5array1T"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1ga3ad3ed7aece2650943a35082dbe3a0a5"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">==</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <a class="reference internal" href="#_CPPv4I0Eeq5arrayRK5array1T" title="operator==::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0Eeq5arrayRK5array1T" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv49not_equalRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv39not_equalRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv29not_equalRK5arrayRK5array14StreamOrDevice"></span><span id="not_equal__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga28f22c5d2c399eee53be7b3facc11103"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">not_equal</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv49not_equalRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns the bool array with (a != b) element-wise. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4neRK5arrayRK5array">
|
|
<span id="_CPPv3neRK5arrayRK5array"></span><span id="_CPPv2neRK5arrayRK5array"></span><span id="neq-operator__arrayCR.arrayCR"></span><span class="target" id="group__ops_1ga0ac483d85f23252ca8757e9926d5a3c5"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">!=</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4neRK5arrayRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0Ene5array1TRK5array">
|
|
<span id="_CPPv3I0Ene1TRK5array"></span><span id="_CPPv2I0Ene1TRK5array"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1ga3fecba9f3cb9a19afd8ca492cf509ce0"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">!=</span></span></span><span class="sig-paren">(</span><a class="reference internal" href="#_CPPv4I0Ene5array1TRK5array" title="operator!=::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0Ene5array1TRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0Ene5arrayRK5array1T">
|
|
<span id="_CPPv3I0EneRK5array1T"></span><span id="_CPPv2I0EneRK5array1T"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1gaebbf1cfde388c7480159a03c92c9a385"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">!=</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <a class="reference internal" href="#_CPPv4I0Ene5arrayRK5array1T" title="operator!=::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0Ene5arrayRK5array1T" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47greaterRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv37greaterRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv27greaterRK5arrayRK5array14StreamOrDevice"></span><span id="greater__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gaf4ec7bfc1ad13b891f1f3ef1772ef04d"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">greater</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47greaterRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns bool array with (a > b) element-wise. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4gtRK5arrayRK5array">
|
|
<span id="_CPPv3gtRK5arrayRK5array"></span><span id="_CPPv2gtRK5arrayRK5array"></span><span id="gt-operator__arrayCR.arrayCR"></span><span class="target" id="group__ops_1ga74fd2777adef10e6fe628a9cdadb01cb"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">></span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4gtRK5arrayRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0Egt5array1TRK5array">
|
|
<span id="_CPPv3I0Egt1TRK5array"></span><span id="_CPPv2I0Egt1TRK5array"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1ga32e106e794e2c32e4e7decee2df2477f"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">></span></span></span><span class="sig-paren">(</span><a class="reference internal" href="#_CPPv4I0Egt5array1TRK5array" title="operator>::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0Egt5array1TRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0Egt5arrayRK5array1T">
|
|
<span id="_CPPv3I0EgtRK5array1T"></span><span id="_CPPv2I0EgtRK5array1T"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1ga96552b90e89923c5d2064cc427775ec5"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">></span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <a class="reference internal" href="#_CPPv4I0Egt5arrayRK5array1T" title="operator>::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0Egt5arrayRK5array1T" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv313greater_equalRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv213greater_equalRK5arrayRK5array14StreamOrDevice"></span><span id="greater_equal__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga7153071bcfff6faad21332163fb9a430"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">greater_equal</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns bool array with (a >= b) element-wise. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4geRK5arrayRK5array">
|
|
<span id="_CPPv3geRK5arrayRK5array"></span><span id="_CPPv2geRK5arrayRK5array"></span><span id="gte-operator__arrayCR.arrayCR"></span><span class="target" id="group__ops_1ga3a41895f25ed083a36994d95fa102546"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">>=</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4geRK5arrayRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0Ege5array1TRK5array">
|
|
<span id="_CPPv3I0Ege1TRK5array"></span><span id="_CPPv2I0Ege1TRK5array"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1gaf509f2cb3b18963232f20d6c3bd229b2"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">>=</span></span></span><span class="sig-paren">(</span><a class="reference internal" href="#_CPPv4I0Ege5array1TRK5array" title="operator>=::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0Ege5array1TRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0Ege5arrayRK5array1T">
|
|
<span id="_CPPv3I0EgeRK5array1T"></span><span id="_CPPv2I0EgeRK5array1T"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1gafa0eb25d5978674bfc9e59d4145ec590"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">>=</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <a class="reference internal" href="#_CPPv4I0Ege5arrayRK5array1T" title="operator>=::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0Ege5arrayRK5array1T" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44lessRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv34lessRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv24lessRK5arrayRK5array14StreamOrDevice"></span><span id="less__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga9142b8d717699a8abfa2a7398891ff8a"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">less</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44lessRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns bool array with (a < b) element-wise. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4ltRK5arrayRK5array">
|
|
<span id="_CPPv3ltRK5arrayRK5array"></span><span id="_CPPv2ltRK5arrayRK5array"></span><span id="lt-operator__arrayCR.arrayCR"></span><span class="target" id="group__ops_1gaee41e2b8f61d563200ff03575ac1d6c3"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre"><</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4ltRK5arrayRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0Elt5array1TRK5array">
|
|
<span id="_CPPv3I0Elt1TRK5array"></span><span id="_CPPv2I0Elt1TRK5array"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1ga1ef8ea11cf15ce628c54201fa42748ef"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre"><</span></span></span><span class="sig-paren">(</span><a class="reference internal" href="#_CPPv4I0Elt5array1TRK5array" title="operator<::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0Elt5array1TRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0Elt5arrayRK5array1T">
|
|
<span id="_CPPv3I0EltRK5array1T"></span><span id="_CPPv2I0EltRK5array1T"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1ga95e72226dc7a79c40b3d16f990922050"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre"><</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <a class="reference internal" href="#_CPPv4I0Elt5arrayRK5array1T" title="operator<::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0Elt5arrayRK5array1T" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv410less_equalRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv310less_equalRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv210less_equalRK5arrayRK5array14StreamOrDevice"></span><span id="less_equal__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga0d49e0c7011d0573c369c13c8f045a09"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">less_equal</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv410less_equalRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns bool array with (a <= b) element-wise. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4leRK5arrayRK5array">
|
|
<span id="_CPPv3leRK5arrayRK5array"></span><span id="_CPPv2leRK5arrayRK5array"></span><span id="lte-operator__arrayCR.arrayCR"></span><span class="target" id="group__ops_1ga4c8b8a1632944acaae50f0de6c23ece6"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre"><=</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4leRK5arrayRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0Ele5array1TRK5array">
|
|
<span id="_CPPv3I0Ele1TRK5array"></span><span id="_CPPv2I0Ele1TRK5array"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1ga150a9be467c9f91482a6d6fc13504bc4"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre"><=</span></span></span><span class="sig-paren">(</span><a class="reference internal" href="#_CPPv4I0Ele5array1TRK5array" title="operator<=::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0Ele5array1TRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0Ele5arrayRK5array1T">
|
|
<span id="_CPPv3I0EleRK5array1T"></span><span id="_CPPv2I0EleRK5array1T"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1ga624eeccef0cc4b130e1325abfea057cb"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre"><=</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <a class="reference internal" href="#_CPPv4I0Ele5arrayRK5array1T" title="operator<=::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0Ele5arrayRK5array1T" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice">
|
|
<span id="_CPPv311array_equalRK5arrayRK5arrayb14StreamOrDevice"></span><span id="_CPPv211array_equalRK5arrayRK5arrayb14StreamOrDevice"></span><span id="array_equal__arrayCR.arrayCR.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga8f3059336ee0c87207b1f8c6ab312645"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">array_equal</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">equal_nan</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>True if two arrays have the same shape and elements. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv411array_equalRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv311array_equalRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv211array_equalRK5arrayRK5array14StreamOrDevice"></span><span id="array_equal__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gaf79cf0271ca0105d7b14295a90d0ed14"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">array_equal</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv411array_equalRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45isnanRK5array14StreamOrDevice">
|
|
<span id="_CPPv35isnanRK5array14StreamOrDevice"></span><span id="_CPPv25isnanRK5array14StreamOrDevice"></span><span id="isnan__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga175592792471b0ffb45196dca4711ba6"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">isnan</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45isnanRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45isinfRK5array14StreamOrDevice">
|
|
<span id="_CPPv35isinfRK5array14StreamOrDevice"></span><span id="_CPPv25isinfRK5array14StreamOrDevice"></span><span id="isinf__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga8fc238d5e5d1153e69da8b36015d9844"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">isinf</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45isinfRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv48isfiniteRK5array14StreamOrDevice">
|
|
<span id="_CPPv38isfiniteRK5array14StreamOrDevice"></span><span id="_CPPv28isfiniteRK5array14StreamOrDevice"></span><span id="isfinite__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga725ff0789f934b1fdd54ee29e47022ff"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">isfinite</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv48isfiniteRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv48isposinfRK5array14StreamOrDevice">
|
|
<span id="_CPPv38isposinfRK5array14StreamOrDevice"></span><span id="_CPPv28isposinfRK5array14StreamOrDevice"></span><span id="isposinf__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gad80f7c4a58c12b6cb30a8b9a73008993"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">isposinf</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv48isposinfRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv48isneginfRK5array14StreamOrDevice">
|
|
<span id="_CPPv38isneginfRK5array14StreamOrDevice"></span><span id="_CPPv28isneginfRK5array14StreamOrDevice"></span><span id="isneginf__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga1940523da381ed7be50656a3bc465ff3"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">isneginf</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv48isneginfRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv35whereRK5arrayRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv25whereRK5arrayRK5arrayRK5array14StreamOrDevice"></span><span id="where__arrayCR.arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga8a2056f8c9bb30914c40bcf509386491"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">where</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">condition</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">x</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">y</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Select from x or y depending on condition. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice">
|
|
<span id="_CPPv310nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice"></span><span id="_CPPv210nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice"></span><span id="nan_to_num__arrayCR.float.std::optional:float:C.std::optional:float:C.StreamOrDevice"></span><span class="target" id="group__ops_1gab1467c6a9e675152e768afd6dcfb61de"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">nan_to_num</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">float</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">nan</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">0.0f</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">optional</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">float</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">posinf</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">nullopt</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">optional</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">float</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">neginf</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">nullopt</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Replace NaN and infinities with finite numbers. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43allRK5arrayb14StreamOrDevice">
|
|
<span id="_CPPv33allRK5arrayb14StreamOrDevice"></span><span id="_CPPv23allRK5arrayb14StreamOrDevice"></span><span id="all__arrayCR.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga3b1b90ef1275ca17655b6d7f25d3ee68"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">all</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43allRK5arrayb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>True if all elements in the array are true (or non-zero). </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43allRK5array14StreamOrDevice">
|
|
<span id="_CPPv33allRK5array14StreamOrDevice"></span><span id="_CPPv23allRK5array14StreamOrDevice"></span><span id="all__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga3689e12e8f42dadb4cbe2b07dc4099f4"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">all</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43allRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice">
|
|
<span id="_CPPv38allcloseRK5arrayRK5arrayddb14StreamOrDevice"></span><span id="_CPPv28allcloseRK5arrayRK5arrayddb14StreamOrDevice"></span><span id="allclose__arrayCR.arrayCR.double.double.b.StreamOrDevice"></span><span class="target" id="group__ops_1gaf0cd4257de7542daf9faf5e605e31020"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">allclose</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="kt"><span class="pre">double</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">rtol</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">1e-5</span></span>, <span class="kt"><span class="pre">double</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">atol</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">1e-8</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">equal_nan</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>True if the two arrays are equal within the specified tolerance. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice">
|
|
<span id="_CPPv37iscloseRK5arrayRK5arrayddb14StreamOrDevice"></span><span id="_CPPv27iscloseRK5arrayRK5arrayddb14StreamOrDevice"></span><span id="isclose__arrayCR.arrayCR.double.double.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga51eac95c04400921c54716de14b52491"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">isclose</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="kt"><span class="pre">double</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">rtol</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">1e-5</span></span>, <span class="kt"><span class="pre">double</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">atol</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">1e-8</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">equal_nan</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns a boolean array where two arrays are element-wise equal within the specified tolerance. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice">
|
|
<span id="_CPPv33allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"></span><span id="_CPPv23allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"></span><span id="all__arrayCR.std::vector:i:CR.b.StreamOrDevice"></span><span class="target" id="group__ops_1gac0919c6ba53aea35a7683dea7e9a9a59"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">all</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Reduces the input along the given axes. </p>
|
|
<p>An output value is true if all the corresponding inputs are true. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43allRK5arrayib14StreamOrDevice">
|
|
<span id="_CPPv33allRK5arrayib14StreamOrDevice"></span><span id="_CPPv23allRK5arrayib14StreamOrDevice"></span><span id="all__arrayCR.i.b.StreamOrDevice"></span><span class="target" id="group__ops_1gae2d5fcc5b62d673cca76c08b7b4afbbc"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">all</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43allRK5arrayib14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Reduces the input along the given axis. </p>
|
|
<p>An output value is true if all the corresponding inputs are true. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43anyRK5arrayb14StreamOrDevice">
|
|
<span id="_CPPv33anyRK5arrayb14StreamOrDevice"></span><span id="_CPPv23anyRK5arrayb14StreamOrDevice"></span><span id="any__arrayCR.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga8598dd718fb05cb28535e250372d4e6f"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">any</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43anyRK5arrayb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>True if any elements in the array are true (or non-zero). </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43anyRK5array14StreamOrDevice">
|
|
<span id="_CPPv33anyRK5array14StreamOrDevice"></span><span id="_CPPv23anyRK5array14StreamOrDevice"></span><span id="any__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gad37df97f253a963bece124198dbaf9ba"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">any</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43anyRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice">
|
|
<span id="_CPPv33anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"></span><span id="_CPPv23anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"></span><span id="any__arrayCR.std::vector:i:CR.b.StreamOrDevice"></span><span class="target" id="group__ops_1gaf240618fc8b06debf5f56e97e84f18ef"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">any</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Reduces the input along the given axes. </p>
|
|
<p>An output value is true if any of the corresponding inputs are true. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43anyRK5arrayib14StreamOrDevice">
|
|
<span id="_CPPv33anyRK5arrayib14StreamOrDevice"></span><span id="_CPPv23anyRK5arrayib14StreamOrDevice"></span><span id="any__arrayCR.i.b.StreamOrDevice"></span><span class="target" id="group__ops_1gab1d56277d468a55227f4dad6bc2fc1ce"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">any</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43anyRK5arrayib14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Reduces the input along the given axis. </p>
|
|
<p>An output value is true if any of the corresponding inputs are true. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43sumRK5arrayb14StreamOrDevice">
|
|
<span id="_CPPv33sumRK5arrayb14StreamOrDevice"></span><span id="_CPPv23sumRK5arrayb14StreamOrDevice"></span><span id="sum__arrayCR.b.StreamOrDevice"></span><span class="target" id="group__ops_1gade905ee92eb6ab7edfc312aeddfbaeb6"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">sum</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43sumRK5arrayb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Sums the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43sumRK5array14StreamOrDevice">
|
|
<span id="_CPPv33sumRK5array14StreamOrDevice"></span><span id="_CPPv23sumRK5array14StreamOrDevice"></span><span id="sum__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga3627754d7868487bdab1bd83f05d9c81"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">sum</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43sumRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice">
|
|
<span id="_CPPv33sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"></span><span id="_CPPv23sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"></span><span id="sum__arrayCR.std::vector:i:CR.b.StreamOrDevice"></span><span class="target" id="group__ops_1gaccd0a6be2c5b5128fdc2d87b5c8e67f4"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">sum</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Sums the elements of an array along the given axes. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43sumRK5arrayib14StreamOrDevice">
|
|
<span id="_CPPv33sumRK5arrayib14StreamOrDevice"></span><span id="_CPPv23sumRK5arrayib14StreamOrDevice"></span><span id="sum__arrayCR.i.b.StreamOrDevice"></span><span class="target" id="group__ops_1gafcd39b0bf39a56c26a967981c7ab8a8d"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">sum</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43sumRK5arrayib14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Sums the elements of an array along the given axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44meanRK5arrayb14StreamOrDevice">
|
|
<span id="_CPPv34meanRK5arrayb14StreamOrDevice"></span><span id="_CPPv24meanRK5arrayb14StreamOrDevice"></span><span id="mean__arrayCR.b.StreamOrDevice"></span><span class="target" id="group__ops_1gade46e768fd46b8b640eb16f26abeecef"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">mean</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44meanRK5arrayb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Computes the mean of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44meanRK5array14StreamOrDevice">
|
|
<span id="_CPPv34meanRK5array14StreamOrDevice"></span><span id="_CPPv24meanRK5array14StreamOrDevice"></span><span id="mean__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga52b59fdd8e8430538e564f5bbcfa31e6"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">mean</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44meanRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice">
|
|
<span id="_CPPv34meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"></span><span id="_CPPv24meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"></span><span id="mean__arrayCR.std::vector:i:CR.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga066161f3d3e395a1d76c638cb680d444"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">mean</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Computes the mean of the elements of an array along the given axes. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44meanRK5arrayib14StreamOrDevice">
|
|
<span id="_CPPv34meanRK5arrayib14StreamOrDevice"></span><span id="_CPPv24meanRK5arrayib14StreamOrDevice"></span><span id="mean__arrayCR.i.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga45fba73eab0e3b6e128ed3ce2f43a5da"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">mean</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44meanRK5arrayib14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Computes the mean of the elements of an array along the given axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43varRK5arraybi14StreamOrDevice">
|
|
<span id="_CPPv33varRK5arraybi14StreamOrDevice"></span><span id="_CPPv23varRK5arraybi14StreamOrDevice"></span><span id="var__arrayCR.b.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga7e133df686439588a8cd1fb10ce0c6e9"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">var</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">ddof</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43varRK5arraybi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Computes the variance of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43varRK5array14StreamOrDevice">
|
|
<span id="_CPPv33varRK5array14StreamOrDevice"></span><span id="_CPPv23varRK5array14StreamOrDevice"></span><span id="var__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga7d7b38d118fa2613214078ef0f7d5a42"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">var</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43varRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice">
|
|
<span id="_CPPv33varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice"></span><span id="_CPPv23varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice"></span><span id="var__arrayCR.std::vector:i:CR.b.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga78ddeb966cbe7a5b0aa17e1de43025f2"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">var</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">ddof</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Computes the variance of the elements of an array along the given axes. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43varRK5arrayibi14StreamOrDevice">
|
|
<span id="_CPPv33varRK5arrayibi14StreamOrDevice"></span><span id="_CPPv23varRK5arrayibi14StreamOrDevice"></span><span id="var__arrayCR.i.b.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga4fbf3e3f98f2e4956faf87af320aa9d0"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">var</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">ddof</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43varRK5arrayibi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Computes the variance of the elements of an array along the given axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4StRK5arraybi14StreamOrDevice">
|
|
<span id="_CPPv3StRK5arraybi14StreamOrDevice"></span><span id="_CPPv2StRK5arraybi14StreamOrDevice"></span><span id="std__arrayCR.b.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga2a466024f8061febc0a64be557644cb0"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">std</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">ddof</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Computes the standard deviation of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4StRK5array14StreamOrDevice">
|
|
<span id="_CPPv3StRK5array14StreamOrDevice"></span><span id="_CPPv2StRK5array14StreamOrDevice"></span><span id="std__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gafdcb04d77c64405a3990078a77dd984c"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">std</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4StRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice">
|
|
<span id="_CPPv3StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice"></span><span id="_CPPv2StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice"></span><span id="std__arrayCR.std::vector:i:CR.b.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga7f649970bf38b987b6ef847054f3c2f8"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">std</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">ddof</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Computes the standard deviation of the elements of an array along the given axes. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4StRK5arrayibi14StreamOrDevice">
|
|
<span id="_CPPv3StRK5arrayibi14StreamOrDevice"></span><span id="_CPPv2StRK5arrayibi14StreamOrDevice"></span><span id="std__arrayCR.i.b.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga62721a206df671ef5797449eea97af9f"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">std</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">ddof</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4StRK5arrayibi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Computes the standard deviation of the elements of an array along the given axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44prodRK5arrayb14StreamOrDevice">
|
|
<span id="_CPPv34prodRK5arrayb14StreamOrDevice"></span><span id="_CPPv24prodRK5arrayb14StreamOrDevice"></span><span id="prod__arrayCR.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga4a09b7241d564d92548bc2773eb1d544"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">prod</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44prodRK5arrayb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>The product of all elements of the array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44prodRK5array14StreamOrDevice">
|
|
<span id="_CPPv34prodRK5array14StreamOrDevice"></span><span id="_CPPv24prodRK5array14StreamOrDevice"></span><span id="prod__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga61832191f3c42ea549cf04953edc3602"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">prod</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44prodRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice">
|
|
<span id="_CPPv34prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"></span><span id="_CPPv24prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"></span><span id="prod__arrayCR.std::vector:i:CR.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga2b3935108f641e20a70dbf63f540d970"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">prod</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>The product of the elements of an array along the given axes. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44prodRK5arrayib14StreamOrDevice">
|
|
<span id="_CPPv34prodRK5arrayib14StreamOrDevice"></span><span id="_CPPv24prodRK5arrayib14StreamOrDevice"></span><span id="prod__arrayCR.i.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga8a10a10b81c69996d0aca8ba401f8ff0"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">prod</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44prodRK5arrayib14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>The product of the elements of an array along the given axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43maxRK5arrayb14StreamOrDevice">
|
|
<span id="_CPPv33maxRK5arrayb14StreamOrDevice"></span><span id="_CPPv23maxRK5arrayb14StreamOrDevice"></span><span id="max__arrayCR.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga7fed87d96cc7741d8267f4eac83f5fe7"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">max</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43maxRK5arrayb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>The maximum of all elements of the array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43maxRK5array14StreamOrDevice">
|
|
<span id="_CPPv33maxRK5array14StreamOrDevice"></span><span id="_CPPv23maxRK5array14StreamOrDevice"></span><span id="max__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga25be91d70a5f40341db0615a0b8bfedc"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">max</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43maxRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice">
|
|
<span id="_CPPv33maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"></span><span id="_CPPv23maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"></span><span id="max__arrayCR.std::vector:i:CR.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga1ca7b6b91fe2459a7d83897bf013827f"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">max</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>The maximum of the elements of an array along the given axes. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43maxRK5arrayib14StreamOrDevice">
|
|
<span id="_CPPv33maxRK5arrayib14StreamOrDevice"></span><span id="_CPPv23maxRK5arrayib14StreamOrDevice"></span><span id="max__arrayCR.i.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga7b638050e03a93f2896c981bc2850a47"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">max</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43maxRK5arrayib14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>The maximum of the elements of an array along the given axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43minRK5arrayb14StreamOrDevice">
|
|
<span id="_CPPv33minRK5arrayb14StreamOrDevice"></span><span id="_CPPv23minRK5arrayb14StreamOrDevice"></span><span id="min__arrayCR.b.StreamOrDevice"></span><span class="target" id="group__ops_1gab27599802617a4c8f9964ab5f4ffee12"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">min</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43minRK5arrayb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>The minimum of all elements of the array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43minRK5array14StreamOrDevice">
|
|
<span id="_CPPv33minRK5array14StreamOrDevice"></span><span id="_CPPv23minRK5array14StreamOrDevice"></span><span id="min__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga0140b91e9cdfc3fef0da8e332f65a9e8"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">min</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43minRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice">
|
|
<span id="_CPPv33minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"></span><span id="_CPPv23minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"></span><span id="min__arrayCR.std::vector:i:CR.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga6efb83cd46436678c8f8c4af15cc00f5"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">min</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>The minimum of the elements of an array along the given axes. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43minRK5arrayib14StreamOrDevice">
|
|
<span id="_CPPv33minRK5arrayib14StreamOrDevice"></span><span id="_CPPv23minRK5arrayib14StreamOrDevice"></span><span id="min__arrayCR.i.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga36fa315eef677f4143868f552cd26d03"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">min</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43minRK5arrayib14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>The minimum of the elements of an array along the given axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46argminRK5arrayb14StreamOrDevice">
|
|
<span id="_CPPv36argminRK5arrayb14StreamOrDevice"></span><span id="_CPPv26argminRK5arrayb14StreamOrDevice"></span><span id="argmin__arrayCR.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga7c3bd5ef430a71dfd298e626741e3c71"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">argmin</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46argminRK5arrayb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns the index of the minimum value in the array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46argminRK5array14StreamOrDevice">
|
|
<span id="_CPPv36argminRK5array14StreamOrDevice"></span><span id="_CPPv26argminRK5array14StreamOrDevice"></span><span id="argmin__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga6bc577c5ab10cd9c848ba81321595070"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">argmin</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46argminRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46argminRK5arrayib14StreamOrDevice">
|
|
<span id="_CPPv36argminRK5arrayib14StreamOrDevice"></span><span id="_CPPv26argminRK5arrayib14StreamOrDevice"></span><span id="argmin__arrayCR.i.b.StreamOrDevice"></span><span class="target" id="group__ops_1gaf66dc3c77b88e4009e0678eda41eca81"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">argmin</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46argminRK5arrayib14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns the indices of the minimum values along a given axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46argmaxRK5arrayb14StreamOrDevice">
|
|
<span id="_CPPv36argmaxRK5arrayb14StreamOrDevice"></span><span id="_CPPv26argmaxRK5arrayb14StreamOrDevice"></span><span id="argmax__arrayCR.b.StreamOrDevice"></span><span class="target" id="group__ops_1gae60b0b5339b9c50b9970260faf613e83"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">argmax</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46argmaxRK5arrayb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns the index of the maximum value in the array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46argmaxRK5array14StreamOrDevice">
|
|
<span id="_CPPv36argmaxRK5array14StreamOrDevice"></span><span id="_CPPv26argmaxRK5array14StreamOrDevice"></span><span id="argmax__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gae6f6c5a840320b336fdc9687e0ed56c8"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">argmax</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46argmaxRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46argmaxRK5arrayib14StreamOrDevice">
|
|
<span id="_CPPv36argmaxRK5arrayib14StreamOrDevice"></span><span id="_CPPv26argmaxRK5arrayib14StreamOrDevice"></span><span id="argmax__arrayCR.i.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga2efa67466510fc26ab9ea8dff30f2ba5"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">argmax</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46argmaxRK5arrayib14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns the indices of the maximum values along a given axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44sortRK5array14StreamOrDevice">
|
|
<span id="_CPPv34sortRK5array14StreamOrDevice"></span><span id="_CPPv24sortRK5array14StreamOrDevice"></span><span id="sort__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga7fb616054665b3c2d61fa234f501f079"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">sort</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44sortRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns a sorted copy of the flattened array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44sortRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv34sortRK5arrayi14StreamOrDevice"></span><span id="_CPPv24sortRK5arrayi14StreamOrDevice"></span><span id="sort__arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1gaae1bc47aa737f705d0e5884270063fea"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">sort</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44sortRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns a sorted copy of the array along a given axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47argsortRK5array14StreamOrDevice">
|
|
<span id="_CPPv37argsortRK5array14StreamOrDevice"></span><span id="_CPPv27argsortRK5array14StreamOrDevice"></span><span id="argsort__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga8df3b2703bf671457422894dd870cdc5"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">argsort</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47argsortRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns indices that sort the flattened array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47argsortRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv37argsortRK5arrayi14StreamOrDevice"></span><span id="_CPPv27argsortRK5arrayi14StreamOrDevice"></span><span id="argsort__arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga7878e0daa5a75f44e57b5fe948fa3ef6"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">argsort</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47argsortRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns indices that sort the array along a given axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv49partitionRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv39partitionRK5arrayi14StreamOrDevice"></span><span id="_CPPv29partitionRK5arrayi14StreamOrDevice"></span><span id="partition__arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1gac1b30830a972fb9a2601379ad2b32405"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">partition</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">kth</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv49partitionRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns a partitioned copy of the flattened array such that the smaller kth elements are first. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv49partitionRK5arrayii14StreamOrDevice">
|
|
<span id="_CPPv39partitionRK5arrayii14StreamOrDevice"></span><span id="_CPPv29partitionRK5arrayii14StreamOrDevice"></span><span id="partition__arrayCR.i.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga4fbea3a5f66cf81e3c119d1661119321"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">partition</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">kth</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv49partitionRK5arrayii14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns a partitioned copy of the array along a given axis such that the smaller kth elements are first. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv412argpartitionRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv312argpartitionRK5arrayi14StreamOrDevice"></span><span id="_CPPv212argpartitionRK5arrayi14StreamOrDevice"></span><span id="argpartition__arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1gaf301c49c10fa9b95a9e8dc52ead1a8dd"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">argpartition</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">kth</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv412argpartitionRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns indices that partition the flattened array such that the smaller kth elements are first. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv412argpartitionRK5arrayii14StreamOrDevice">
|
|
<span id="_CPPv312argpartitionRK5arrayii14StreamOrDevice"></span><span id="_CPPv212argpartitionRK5arrayii14StreamOrDevice"></span><span id="argpartition__arrayCR.i.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga7b15c654c7463def57857a0e239989a3"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">argpartition</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">kth</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv412argpartitionRK5arrayii14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns indices that partition the array along a given axis such that the smaller kth elements are first. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44topkRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv34topkRK5arrayi14StreamOrDevice"></span><span id="_CPPv24topkRK5arrayi14StreamOrDevice"></span><span id="topk__arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga5487dd887c43e5341f3e68ffe47f0f5a"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">topk</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">k</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44topkRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns topk elements of the flattened array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44topkRK5arrayii14StreamOrDevice">
|
|
<span id="_CPPv34topkRK5arrayii14StreamOrDevice"></span><span id="_CPPv24topkRK5arrayii14StreamOrDevice"></span><span id="topk__arrayCR.i.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga35b8436c79ff953f6c809598b646f498"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">topk</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">k</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44topkRK5arrayii14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns topk elements of the array along a given axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv412logcumsumexpRK5arrayibb14StreamOrDevice">
|
|
<span id="_CPPv312logcumsumexpRK5arrayibb14StreamOrDevice"></span><span id="_CPPv212logcumsumexpRK5arrayibb14StreamOrDevice"></span><span id="logcumsumexp__arrayCR.i.b.b.StreamOrDevice"></span><span class="target" id="group__ops_1gaeb541012d52d7082365048f9e094cae4"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">logcumsumexp</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">reverse</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">inclusive</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">true</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv412logcumsumexpRK5arrayibb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Cumulative logsumexp of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv49logsumexpRK5arrayb14StreamOrDevice">
|
|
<span id="_CPPv39logsumexpRK5arrayb14StreamOrDevice"></span><span id="_CPPv29logsumexpRK5arrayb14StreamOrDevice"></span><span id="logsumexp__arrayCR.b.StreamOrDevice"></span><span class="target" id="group__ops_1gacff4eb57c085d571e722083680267ac5"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">logsumexp</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv49logsumexpRK5arrayb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>The logsumexp of all elements of the array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv49logsumexpRK5array14StreamOrDevice">
|
|
<span id="_CPPv39logsumexpRK5array14StreamOrDevice"></span><span id="_CPPv29logsumexpRK5array14StreamOrDevice"></span><span id="logsumexp__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga59be50b4e92f1dc20b53460cefa3910d"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">logsumexp</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv49logsumexpRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice">
|
|
<span id="_CPPv39logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"></span><span id="_CPPv29logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"></span><span id="logsumexp__arrayCR.std::vector:i:CR.b.StreamOrDevice"></span><span class="target" id="group__ops_1gae3969c7bd24c4f3ab97831df28239689"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">logsumexp</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>The logsumexp of the elements of an array along the given axes. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv49logsumexpRK5arrayib14StreamOrDevice">
|
|
<span id="_CPPv39logsumexpRK5arrayib14StreamOrDevice"></span><span id="_CPPv29logsumexpRK5arrayib14StreamOrDevice"></span><span id="logsumexp__arrayCR.i.b.StreamOrDevice"></span><span class="target" id="group__ops_1gafef5cb2159c16a60a95470cc823bdd44"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">logsumexp</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">keepdims</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv49logsumexpRK5arrayib14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>The logsumexp of the elements of an array along the given axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43absRK5array14StreamOrDevice">
|
|
<span id="_CPPv33absRK5array14StreamOrDevice"></span><span id="_CPPv23absRK5array14StreamOrDevice"></span><span id="abs__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga5528e80f5e8bad71e106a0cf9edd8920"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">abs</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43absRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Absolute value of elements in an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv48negativeRK5array14StreamOrDevice">
|
|
<span id="_CPPv38negativeRK5array14StreamOrDevice"></span><span id="_CPPv28negativeRK5array14StreamOrDevice"></span><span id="negative__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga95d9a9425533b5ed1707eb00184dffc6"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">negative</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv48negativeRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Negate an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4miRK5array">
|
|
<span id="_CPPv3miRK5array"></span><span id="_CPPv2miRK5array"></span><span id="sub-operator__arrayCR"></span><span class="target" id="group__ops_1gade2eea48989f4caaf36e89f7bd2a8816"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">-</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4miRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44signRK5array14StreamOrDevice">
|
|
<span id="_CPPv34signRK5array14StreamOrDevice"></span><span id="_CPPv24signRK5array14StreamOrDevice"></span><span id="sign__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga20f1a1a8c0cd6206485f9363f3915faa"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">sign</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44signRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>The sign of the elements in an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv411logical_notRK5array14StreamOrDevice">
|
|
<span id="_CPPv311logical_notRK5array14StreamOrDevice"></span><span id="_CPPv211logical_notRK5array14StreamOrDevice"></span><span id="logical_not__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gabca78d34ce93f0de2814e62225bb2a53"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">logical_not</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv411logical_notRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Logical not of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv411logical_andRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv311logical_andRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv211logical_andRK5arrayRK5array14StreamOrDevice"></span><span id="logical_and__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga768977cda8d68cf23f464a6af9907876"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">logical_and</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv411logical_andRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Logical and of two arrays. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4aaRK5arrayRK5array">
|
|
<span id="_CPPv3aaRK5arrayRK5array"></span><span id="_CPPv2aaRK5arrayRK5array"></span><span id="sand-operator__arrayCR.arrayCR"></span><span class="target" id="group__ops_1gaee1d774bb0843601d7a0a4257d616ae3"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">&&</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4aaRK5arrayRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv410logical_orRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv310logical_orRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv210logical_orRK5arrayRK5array14StreamOrDevice"></span><span id="logical_or__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga23768728e4dd070c917fbb0ed0d0c2ec"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">logical_or</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv410logical_orRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Logical or of two arrays. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4ooRK5arrayRK5array">
|
|
<span id="_CPPv3ooRK5arrayRK5array"></span><span id="_CPPv2ooRK5arrayRK5array"></span><span id="sor-operator__arrayCR.arrayCR"></span><span class="target" id="group__ops_1ga27af56a98270d4d76d139f0f9171b83a"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">||</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4ooRK5arrayRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv410reciprocalRK5array14StreamOrDevice">
|
|
<span id="_CPPv310reciprocalRK5array14StreamOrDevice"></span><span id="_CPPv210reciprocalRK5array14StreamOrDevice"></span><span id="reciprocal__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga4d29556bb93e2f66916116cf1f062b36"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">reciprocal</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv410reciprocalRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>The reciprocal (1/x) of the elements in an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43addRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv33addRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv23addRK5arrayRK5array14StreamOrDevice"></span><span id="add__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga2d32d67cfd76785a72c43d89b94dc7d7"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">add</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43addRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Add two arrays. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4plRK5arrayRK5array">
|
|
<span id="_CPPv3plRK5arrayRK5array"></span><span id="_CPPv2plRK5arrayRK5array"></span><span id="add-operator__arrayCR.arrayCR"></span><span class="target" id="group__ops_1ga26e5a043eaaaf066d1400adac9c11d0c"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">+</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4plRK5arrayRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0Epl5array1TRK5array">
|
|
<span id="_CPPv3I0Epl1TRK5array"></span><span id="_CPPv2I0Epl1TRK5array"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1ga7d0ec8d01e7cefa6a6b25f11876761b5"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">+</span></span></span><span class="sig-paren">(</span><a class="reference internal" href="#_CPPv4I0Epl5array1TRK5array" title="operator+::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0Epl5array1TRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0Epl5arrayRK5array1T">
|
|
<span id="_CPPv3I0EplRK5array1T"></span><span id="_CPPv2I0EplRK5array1T"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1ga7cc080a4f9d4a667f2099aa0dbfefadd"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">+</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <a class="reference internal" href="#_CPPv4I0Epl5arrayRK5array1T" title="operator+::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0Epl5arrayRK5array1T" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv48subtractRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv38subtractRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv28subtractRK5arrayRK5array14StreamOrDevice"></span><span id="subtract__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga196c240d3d0fcbb4713802c485e15133"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">subtract</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv48subtractRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Subtract two arrays. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4miRK5arrayRK5array">
|
|
<span id="_CPPv3miRK5arrayRK5array"></span><span id="_CPPv2miRK5arrayRK5array"></span><span id="sub-operator__arrayCR.arrayCR"></span><span class="target" id="group__ops_1ga0c7f3cb36d4ca516c7a33142f88b9181"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">-</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4miRK5arrayRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0Emi5array1TRK5array">
|
|
<span id="_CPPv3I0Emi1TRK5array"></span><span id="_CPPv2I0Emi1TRK5array"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1gae68d3d0691ba951501218e98439f3465"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">-</span></span></span><span class="sig-paren">(</span><a class="reference internal" href="#_CPPv4I0Emi5array1TRK5array" title="operator-::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0Emi5array1TRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0Emi5arrayRK5array1T">
|
|
<span id="_CPPv3I0EmiRK5array1T"></span><span id="_CPPv2I0EmiRK5array1T"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1gaf5e5d882c51ad0a0ea315c274d5439b2"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">-</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <a class="reference internal" href="#_CPPv4I0Emi5arrayRK5array1T" title="operator-::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0Emi5arrayRK5array1T" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv48multiplyRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv38multiplyRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv28multiplyRK5arrayRK5array14StreamOrDevice"></span><span id="multiply__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gaf57392e641640b5d06e4c99518391c38"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">multiply</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv48multiplyRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Multiply two arrays. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4mlRK5arrayRK5array">
|
|
<span id="_CPPv3mlRK5arrayRK5array"></span><span id="_CPPv2mlRK5arrayRK5array"></span><span id="mul-operator__arrayCR.arrayCR"></span><span class="target" id="group__ops_1ga26c33f5cdb6fc10d272acd6e208034e0"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">*</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4mlRK5arrayRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0Eml5array1TRK5array">
|
|
<span id="_CPPv3I0Eml1TRK5array"></span><span id="_CPPv2I0Eml1TRK5array"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1gac22a67f7de797b1ae59029843cbdcab6"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">*</span></span></span><span class="sig-paren">(</span><a class="reference internal" href="#_CPPv4I0Eml5array1TRK5array" title="operator*::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0Eml5array1TRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0Eml5arrayRK5array1T">
|
|
<span id="_CPPv3I0EmlRK5array1T"></span><span id="_CPPv2I0EmlRK5array1T"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1ga6f2369ed5fae8ff9b1528670a004dde2"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">*</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <a class="reference internal" href="#_CPPv4I0Eml5arrayRK5array1T" title="operator*::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0Eml5arrayRK5array1T" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46divideRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv36divideRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv26divideRK5arrayRK5array14StreamOrDevice"></span><span id="divide__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga77472dd06cfa7a30a42e4fd927bd859f"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">divide</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46divideRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Divide two arrays. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4dvRK5arrayRK5array">
|
|
<span id="_CPPv3dvRK5arrayRK5array"></span><span id="_CPPv2dvRK5arrayRK5array"></span><span id="div-operator__arrayCR.arrayCR"></span><span class="target" id="group__ops_1gaeedf77f722b394429f1a7f6c367883bf"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">/</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4dvRK5arrayRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4dvdRK5array">
|
|
<span id="_CPPv3dvdRK5array"></span><span id="_CPPv2dvdRK5array"></span><span id="div-operator__double.arrayCR"></span><span class="target" id="group__ops_1ga7366ec7f453be2a4dc449f0faa1bf554"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">/</span></span></span><span class="sig-paren">(</span><span class="kt"><span class="pre">double</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4dvdRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4dvRK5arrayd">
|
|
<span id="_CPPv3dvRK5arrayd"></span><span id="_CPPv2dvRK5arrayd"></span><span id="div-operator__arrayCR.double"></span><span class="target" id="group__ops_1gadfb324ae9b4feb2c7ea0ac6ade639f38"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">/</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">double</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4dvRK5arrayd" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46divmodRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv36divmodRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv26divmodRK5arrayRK5array14StreamOrDevice"></span><span id="divmod__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gaa30ebc0a8376dbc3f7e46a47052b5894"></span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">divmod</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46divmodRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Compute the element-wise quotient and remainder. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv312floor_divideRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv212floor_divideRK5arrayRK5array14StreamOrDevice"></span><span id="floor_divide__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga05b4c6054d028107869511f927da01cd"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">floor_divide</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Compute integer division. </p>
|
|
<p>Equivalent to doing floor(a / x). </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv49remainderRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv39remainderRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv29remainderRK5arrayRK5array14StreamOrDevice"></span><span id="remainder__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga99f5c904f724156a814d7817188351d2"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">remainder</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv49remainderRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Compute the element-wise remainder of division. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4rmRK5arrayRK5array">
|
|
<span id="_CPPv3rmRK5arrayRK5array"></span><span id="_CPPv2rmRK5arrayRK5array"></span><span id="mod-operator__arrayCR.arrayCR"></span><span class="target" id="group__ops_1gab3bfbf82b1e4de7b00bbcf1a2255fbde"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">%</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4rmRK5arrayRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0Erm5array1TRK5array">
|
|
<span id="_CPPv3I0Erm1TRK5array"></span><span id="_CPPv2I0Erm1TRK5array"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1ga50817666f0b82afcbf4a123486af9908"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">%</span></span></span><span class="sig-paren">(</span><a class="reference internal" href="#_CPPv4I0Erm5array1TRK5array" title="operator%::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0Erm5array1TRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4I0Erm5arrayRK5array1T">
|
|
<span id="_CPPv3I0ErmRK5array1T"></span><span id="_CPPv2I0ErmRK5array1T"></span><span class="k"><span class="pre">template</span></span><span class="p"><span class="pre"><</span></span><span class="k"><span class="pre">typename</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">T</span></span></span><span class="p"><span class="pre">></span></span><br /><span class="target" id="group__ops_1ga46c01daa07433542a477d216e13a8480"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">%</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <a class="reference internal" href="#_CPPv4I0Erm5arrayRK5array1T" title="operator%::T"><span class="n"><span class="pre">T</span></span></a><span class="w"> </span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4I0Erm5arrayRK5array1T" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47maximumRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv37maximumRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv27maximumRK5arrayRK5array14StreamOrDevice"></span><span id="maximum__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga7ade2ea305e2e4219c3609443fb5db8d"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">maximum</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47maximumRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Element-wise maximum between two arrays. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47minimumRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv37minimumRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv27minimumRK5arrayRK5array14StreamOrDevice"></span><span id="minimum__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga49ba00c090f81f331c91b0c97040bce0"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">minimum</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47minimumRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Element-wise minimum between two arrays. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45floorRK5array14StreamOrDevice">
|
|
<span id="_CPPv35floorRK5array14StreamOrDevice"></span><span id="_CPPv25floorRK5array14StreamOrDevice"></span><span id="floor__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga8d656904aa2690b60955ae745aecfc30"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">floor</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45floorRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Floor the element of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44ceilRK5array14StreamOrDevice">
|
|
<span id="_CPPv34ceilRK5array14StreamOrDevice"></span><span id="_CPPv24ceilRK5array14StreamOrDevice"></span><span id="ceil__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga1404ecceff83fd9b9139b7520f55e096"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">ceil</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44ceilRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Ceil the element of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46squareRK5array14StreamOrDevice">
|
|
<span id="_CPPv36squareRK5array14StreamOrDevice"></span><span id="_CPPv26squareRK5array14StreamOrDevice"></span><span id="square__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga1234e4c39cfa79f19d4bdb5b8ea4d45e"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">square</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46squareRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Square the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43expRK5array14StreamOrDevice">
|
|
<span id="_CPPv33expRK5array14StreamOrDevice"></span><span id="_CPPv23expRK5array14StreamOrDevice"></span><span id="exp__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga8a3b04e23e347d99ecf411fd6f4e5125"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">exp</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43expRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Exponential of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43sinRK5array14StreamOrDevice">
|
|
<span id="_CPPv33sinRK5array14StreamOrDevice"></span><span id="_CPPv23sinRK5array14StreamOrDevice"></span><span id="sin__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gaebf0a73ad3732fba39df37826c235692"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">sin</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43sinRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Sine of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43cosRK5array14StreamOrDevice">
|
|
<span id="_CPPv33cosRK5array14StreamOrDevice"></span><span id="_CPPv23cosRK5array14StreamOrDevice"></span><span id="cos__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga39dfdf72b556012aa35ff27a94116e74"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">cos</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43cosRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Cosine of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43tanRK5array14StreamOrDevice">
|
|
<span id="_CPPv33tanRK5array14StreamOrDevice"></span><span id="_CPPv23tanRK5array14StreamOrDevice"></span><span id="tan__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga3f10e89a4bcb1a8fa44fb33b8d1176a5"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">tan</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43tanRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Tangent of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46arcsinRK5array14StreamOrDevice">
|
|
<span id="_CPPv36arcsinRK5array14StreamOrDevice"></span><span id="_CPPv26arcsinRK5array14StreamOrDevice"></span><span id="arcsin__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga8770e8c8f23f13343911f4c9d6e1c619"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">arcsin</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46arcsinRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Arc Sine of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46arccosRK5array14StreamOrDevice">
|
|
<span id="_CPPv36arccosRK5array14StreamOrDevice"></span><span id="_CPPv26arccosRK5array14StreamOrDevice"></span><span id="arccos__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga08bec7cb10c84466487b507fc5bf9776"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">arccos</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46arccosRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Arc Cosine of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46arctanRK5array14StreamOrDevice">
|
|
<span id="_CPPv36arctanRK5array14StreamOrDevice"></span><span id="_CPPv26arctanRK5array14StreamOrDevice"></span><span id="arctan__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gaa041f3f070e68f4946db07516b7d092e"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">arctan</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46arctanRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Arc Tangent of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47arctan2RK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv37arctan2RK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv27arctan2RK5arrayRK5array14StreamOrDevice"></span><span id="arctan2__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga6caba9c92b5989123501f909cc7da354"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">arctan2</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47arctan2RK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Inverse tangent of the ratio of two arrays. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44sinhRK5array14StreamOrDevice">
|
|
<span id="_CPPv34sinhRK5array14StreamOrDevice"></span><span id="_CPPv24sinhRK5array14StreamOrDevice"></span><span id="sinh__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gaf532375c6563dbd6e329bdedf0224dd7"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">sinh</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44sinhRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Hyperbolic Sine of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44coshRK5array14StreamOrDevice">
|
|
<span id="_CPPv34coshRK5array14StreamOrDevice"></span><span id="_CPPv24coshRK5array14StreamOrDevice"></span><span id="cosh__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga2181b71cda88007a3092be4795ff0715"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">cosh</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44coshRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Hyperbolic Cosine of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44tanhRK5array14StreamOrDevice">
|
|
<span id="_CPPv34tanhRK5array14StreamOrDevice"></span><span id="_CPPv24tanhRK5array14StreamOrDevice"></span><span id="tanh__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga5efb19aa0dfa42d8a3d5e1dfd569cd6d"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">tanh</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44tanhRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Hyperbolic Tangent of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47arcsinhRK5array14StreamOrDevice">
|
|
<span id="_CPPv37arcsinhRK5array14StreamOrDevice"></span><span id="_CPPv27arcsinhRK5array14StreamOrDevice"></span><span id="arcsinh__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gac62e2cedc49ef2c90dd8584000317450"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">arcsinh</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47arcsinhRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Inverse Hyperbolic Sine of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47arccoshRK5array14StreamOrDevice">
|
|
<span id="_CPPv37arccoshRK5array14StreamOrDevice"></span><span id="_CPPv27arccoshRK5array14StreamOrDevice"></span><span id="arccosh__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gaafafcfcebdf7248679c8543d0c0497e5"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">arccosh</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47arccoshRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Inverse Hyperbolic Cosine of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47arctanhRK5array14StreamOrDevice">
|
|
<span id="_CPPv37arctanhRK5array14StreamOrDevice"></span><span id="_CPPv27arctanhRK5array14StreamOrDevice"></span><span id="arctanh__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gab46a35925a04c5a9d2ec7898ee55358e"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">arctanh</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47arctanhRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Inverse Hyperbolic Tangent of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47degreesRK5array14StreamOrDevice">
|
|
<span id="_CPPv37degreesRK5array14StreamOrDevice"></span><span id="_CPPv27degreesRK5array14StreamOrDevice"></span><span id="degrees__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga3a70569b50e1083c5ded199d73fb960c"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">degrees</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47degreesRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Convert the elements of an array from Radians to Degrees. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47radiansRK5array14StreamOrDevice">
|
|
<span id="_CPPv37radiansRK5array14StreamOrDevice"></span><span id="_CPPv27radiansRK5array14StreamOrDevice"></span><span id="radians__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga07aa8059adba5b9a8818027b8aafd31e"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">radians</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47radiansRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Convert the elements of an array from Degrees to Radians. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43logRK5array14StreamOrDevice">
|
|
<span id="_CPPv33logRK5array14StreamOrDevice"></span><span id="_CPPv23logRK5array14StreamOrDevice"></span><span id="log__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga6fb22d4926133573e430fcc92f4eef31"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">log</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43logRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Natural logarithm of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44log2RK5array14StreamOrDevice">
|
|
<span id="_CPPv34log2RK5array14StreamOrDevice"></span><span id="_CPPv24log2RK5array14StreamOrDevice"></span><span id="log2__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga144228d7222d15af3a135b8b0f3fa21b"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">log2</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44log2RK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Log base 2 of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45log10RK5array14StreamOrDevice">
|
|
<span id="_CPPv35log10RK5array14StreamOrDevice"></span><span id="_CPPv25log10RK5array14StreamOrDevice"></span><span id="log10__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga1fdcc7fc8819caf2e6f1c327ed4e9b9e"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">log10</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45log10RK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Log base 10 of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45log1pRK5array14StreamOrDevice">
|
|
<span id="_CPPv35log1pRK5array14StreamOrDevice"></span><span id="_CPPv25log1pRK5array14StreamOrDevice"></span><span id="log1p__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga20a1f4270c35b0fa544f5105a87a1604"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">log1p</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45log1pRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Natural logarithm of one plus elements in the array: <code class="docutils literal notranslate"><span class="pre">log(1</span> <span class="pre">+</span> <span class="pre">a)</span></code>. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv39logaddexpRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv29logaddexpRK5arrayRK5array14StreamOrDevice"></span><span id="logaddexp__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gaf985df6609c6bd75a14a844655d89eaa"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">logaddexp</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Log-add-exp of one elements in the array: <code class="docutils literal notranslate"><span class="pre">log(exp(a)</span> <span class="pre">+</span> <span class="pre">exp(b))</span></code>. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47sigmoidRK5array14StreamOrDevice">
|
|
<span id="_CPPv37sigmoidRK5array14StreamOrDevice"></span><span id="_CPPv27sigmoidRK5array14StreamOrDevice"></span><span id="sigmoid__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga708abf8f79609cd6831db7c38cafac0e"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">sigmoid</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47sigmoidRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Element-wise logistic sigmoid of the array: <code class="docutils literal notranslate"><span class="pre">1</span> <span class="pre">/</span> <span class="pre">(1</span> <span class="pre">+</span> <span class="pre">exp(-x)</span></code>. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv43erfRK5array14StreamOrDevice">
|
|
<span id="_CPPv33erfRK5array14StreamOrDevice"></span><span id="_CPPv23erfRK5array14StreamOrDevice"></span><span id="erf__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga292a335240fd5d6d625fb7a340ff5eb0"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">erf</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv43erfRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Computes the error function of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46erfinvRK5array14StreamOrDevice">
|
|
<span id="_CPPv36erfinvRK5array14StreamOrDevice"></span><span id="_CPPv26erfinvRK5array14StreamOrDevice"></span><span id="erfinv__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga76fb9062c64264e34d2e07013390557c"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">erfinv</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46erfinvRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Computes the inverse error function of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45expm1RK5array14StreamOrDevice">
|
|
<span id="_CPPv35expm1RK5array14StreamOrDevice"></span><span id="_CPPv25expm1RK5array14StreamOrDevice"></span><span id="expm1__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga54ca54f06bfb2be15b163a5209e2a0f0"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">expm1</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45expm1RK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Computes the expm1 function of the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv413stop_gradientRK5array14StreamOrDevice">
|
|
<span id="_CPPv313stop_gradientRK5array14StreamOrDevice"></span><span id="_CPPv213stop_gradientRK5array14StreamOrDevice"></span><span id="stop_gradient__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga36bc28f1deb2fe668ca9ae1e447b6b1f"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">stop_gradient</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv413stop_gradientRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Stop the flow of gradients. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45roundRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv35roundRK5arrayi14StreamOrDevice"></span><span id="_CPPv25roundRK5arrayi14StreamOrDevice"></span><span id="round__arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga2d74d43f007a069384e89d8416525331"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">round</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">decimals</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45roundRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Round a floating point number. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45roundRK5array14StreamOrDevice">
|
|
<span id="_CPPv35roundRK5array14StreamOrDevice"></span><span id="_CPPv25roundRK5array14StreamOrDevice"></span><span id="round__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gaf18fb7e98bf8cf3b7fbc5e64c988a95b"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">round</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45roundRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46matmulRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv36matmulRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv26matmulRK5arrayRK5array14StreamOrDevice"></span><span id="matmul__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga753d59f5a9f5f2362865ee83b4dced2a"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">matmul</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46matmulRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Matrix-matrix multiplication. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice">
|
|
<span id="_CPPv36gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice"></span><span id="_CPPv26gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice"></span><span id="gather__arrayCR.std::vector:array:CR.std::vector:i:CR.ShapeCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga8fcc3ad0677c834c36b72d5b2ebba6d0"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">gather</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">indices</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">slice_sizes</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Gather array entries given indices and slices. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice">
|
|
<span id="_CPPv36gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice"></span><span id="_CPPv26gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice"></span><span id="gather__arrayCR.arrayCR.i.ShapeCR.StreamOrDevice"></span><span class="target" id="group__ops_1gafe2bd174c9953ed7f12664f7abaca0e6"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">gather</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">indices</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">slice_sizes</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44kronRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv34kronRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv24kronRK5arrayRK5array14StreamOrDevice"></span><span id="kron__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga6df16248cb68bc73644cdb1698967c19"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">kron</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44kronRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Compute the Kronecker product of two arrays. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv34takeRK5arrayRK5arrayi14StreamOrDevice"></span><span id="_CPPv24takeRK5arrayRK5arrayi14StreamOrDevice"></span><span id="take__arrayCR.arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1gac2fc270882fcfa81eb8bd068cc0d86d7"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">take</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">indices</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Take array slices at the given indices of the specified axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44takeRK5arrayii14StreamOrDevice">
|
|
<span id="_CPPv34takeRK5arrayii14StreamOrDevice"></span><span id="_CPPv24takeRK5arrayii14StreamOrDevice"></span><span id="take__arrayCR.i.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga731af77b7de547a73336f91c829c40b4"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">take</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">index</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44takeRK5arrayii14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44takeRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv34takeRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv24takeRK5arrayRK5array14StreamOrDevice"></span><span id="take__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga45d0f423a5e030440ef753f36c5aabf1"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">take</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">indices</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44takeRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Take array entries at the given indices treating the array as flattened. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44takeRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv34takeRK5arrayi14StreamOrDevice"></span><span id="_CPPv24takeRK5arrayi14StreamOrDevice"></span><span id="take__arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga7b3c9b4c8ee02dc23cfd7dcd855dbf20"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">take</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">index</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44takeRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv315take_along_axisRK5arrayRK5arrayi14StreamOrDevice"></span><span id="_CPPv215take_along_axisRK5arrayRK5arrayi14StreamOrDevice"></span><span id="take_along_axis__arrayCR.arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1gae0a81d4f983e296a87302e36d65bfc76"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">take_along_axis</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">indices</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Take array entries given indices along the axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv314put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice"></span><span id="_CPPv214put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice"></span><span id="put_along_axis__arrayCR.arrayCR.arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga8e0caebf43cd65bd40e4ce97922cd06b"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">put_along_axis</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">indices</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">values</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Put the values into the array at the given indices along the axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv316scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice"></span><span id="_CPPv216scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice"></span><span id="scatter_add_axis__arrayCR.arrayCR.arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1gab3fd98c0d06b84b836f93bddbd7a2a0d"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">scatter_add_axis</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">indices</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">values</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Add the values into the array at the given indices along the axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice">
|
|
<span id="_CPPv37scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"></span><span id="_CPPv27scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"></span><span id="scatter__arrayCR.std::vector:array:CR.arrayCR.std::vector:i:CR.StreamOrDevice"></span><span class="target" id="group__ops_1gad438be8f90bae9d37c6853b8f4225d61"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">scatter</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">indices</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">updates</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Scatter updates to the given indices. </p>
|
|
<p>The parameters <code class="docutils literal notranslate"><span class="pre">indices</span></code> and <code class="docutils literal notranslate"><span class="pre">axes</span></code> determine the locations of <code class="docutils literal notranslate"><span class="pre">a</span></code> that are updated with the values in <code class="docutils literal notranslate"><span class="pre">updates</span></code>. Assuming 1-d <code class="docutils literal notranslate"><span class="pre">indices</span></code> for simplicity, <code class="docutils literal notranslate"><span class="pre">indices[i]</span></code> are the indices on axis <code class="docutils literal notranslate"><span class="pre">axes[i]</span></code> to which the values in <code class="docutils literal notranslate"><span class="pre">updates</span></code> will be applied. Note each array in <code class="docutils literal notranslate"><span class="pre">indices</span></code> is assigned to a corresponding axis and hence <code class="docutils literal notranslate"><span class="pre">indices.size()</span> <span class="pre">==</span> <span class="pre">axes.size()</span></code>. If an index/axis pair is not provided then indices along that axis are assumed to be zero.</p>
|
|
<p>Note the rank of <code class="docutils literal notranslate"><span class="pre">updates</span></code> must be equal to the sum of the rank of the broadcasted <code class="docutils literal notranslate"><span class="pre">indices</span></code> and the rank of <code class="docutils literal notranslate"><span class="pre">a</span></code>. In other words, assuming the arrays in <code class="docutils literal notranslate"><span class="pre">indices</span></code> have the same shape, <code class="docutils literal notranslate"><span class="pre">updates.ndim()</span> <span class="pre">==</span> <span class="pre">indices[0].ndim()</span> <span class="pre">+</span> <span class="pre">a.ndim()</span></code>. The leading dimensions of <code class="docutils literal notranslate"><span class="pre">updates</span></code> correspond to the indices, and the remaining <code class="docutils literal notranslate"><span class="pre">a.ndim()</span></code> dimensions are the values that will be applied to the given location in <code class="docutils literal notranslate"><span class="pre">a</span></code>.</p>
|
|
<p>For example:</p>
|
|
<div class="highlight-python notranslate"><div class="highlight"><pre><span></span><span class="n">auto</span> <span class="ow">in</span> <span class="o">=</span> <span class="n">zeros</span><span class="p">({</span><span class="mi">4</span><span class="p">,</span> <span class="mi">4</span><span class="p">},</span> <span class="n">float32</span><span class="p">);</span>
|
|
<span class="n">auto</span> <span class="n">indices</span> <span class="o">=</span> <span class="n">array</span><span class="p">({</span><span class="mi">2</span><span class="p">});</span>
|
|
<span class="n">auto</span> <span class="n">updates</span> <span class="o">=</span> <span class="n">reshape</span><span class="p">(</span><span class="n">arange</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span class="mi">3</span><span class="p">,</span> <span class="n">float32</span><span class="p">),</span> <span class="p">{</span><span class="mi">1</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">});</span>
|
|
<span class="n">std</span><span class="p">::</span><span class="n">vector</span><span class="o"><</span><span class="nb">int</span><span class="o">></span> <span class="n">axes</span><span class="p">{</span><span class="mi">0</span><span class="p">};</span>
|
|
|
|
<span class="n">auto</span> <span class="n">out</span> <span class="o">=</span> <span class="n">scatter</span><span class="p">(</span><span class="ow">in</span><span class="p">,</span> <span class="p">{</span><span class="n">indices</span><span class="p">},</span> <span class="n">updates</span><span class="p">,</span> <span class="n">axes</span><span class="p">);</span>
|
|
</pre></div>
|
|
</div>
|
|
<p>will produce:</p>
|
|
<div class="highlight-python notranslate"><div class="highlight"><pre><span></span><span class="n">array</span><span class="p">([[</span><span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">],</span>
|
|
<span class="p">[</span><span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">],</span>
|
|
<span class="p">[</span><span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">],</span>
|
|
<span class="p">[</span><span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">]],</span> <span class="n">dtype</span><span class="o">=</span><span class="n">float32</span><span class="p">)</span>
|
|
</pre></div>
|
|
</div>
|
|
<p>This scatters the two-element row vector <code class="docutils literal notranslate"><span class="pre">[1,</span> <span class="pre">2]</span></code> starting at the <code class="docutils literal notranslate"><span class="pre">(2,</span> <span class="pre">0)</span></code> position of <code class="docutils literal notranslate"><span class="pre">a</span></code>.</p>
|
|
<p>Adding another element to <code class="docutils literal notranslate"><span class="pre">indices</span></code> will scatter into another location of <code class="docutils literal notranslate"><span class="pre">a</span></code>. We also have to add an another update for the new index:</p>
|
|
<div class="highlight-python notranslate"><div class="highlight"><pre><span></span><span class="n">auto</span> <span class="ow">in</span> <span class="o">=</span> <span class="n">zeros</span><span class="p">({</span><span class="mi">4</span><span class="p">,</span> <span class="mi">4</span><span class="p">},</span> <span class="n">float32</span><span class="p">);</span>
|
|
<span class="n">auto</span> <span class="n">indices</span> <span class="o">=</span> <span class="n">array</span><span class="p">({</span><span class="mi">2</span><span class="p">,</span> <span class="mi">0</span><span class="p">});</span>
|
|
<span class="n">auto</span> <span class="n">updates</span> <span class="o">=</span> <span class="n">reshape</span><span class="p">(</span><span class="n">arange</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span class="mi">5</span><span class="p">,</span> <span class="n">float32</span><span class="p">),</span> <span class="p">{</span><span class="mi">2</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">});</span>
|
|
<span class="n">std</span><span class="p">::</span><span class="n">vector</span><span class="o"><</span><span class="nb">int</span><span class="o">></span> <span class="n">axes</span><span class="p">{</span><span class="mi">0</span><span class="p">};</span>
|
|
|
|
<span class="n">auto</span> <span class="n">out</span> <span class="o">=</span> <span class="n">scatter</span><span class="p">(</span><span class="ow">in</span><span class="p">,</span> <span class="p">{</span><span class="n">indices</span><span class="p">},</span> <span class="n">updates</span><span class="p">,</span> <span class="n">axes</span><span class="p">):</span>
|
|
</pre></div>
|
|
</div>
|
|
<p>will produce:</p>
|
|
<div class="highlight-python notranslate"><div class="highlight"><pre><span></span><span class="n">array</span><span class="p">([[</span><span class="mi">3</span><span class="p">,</span> <span class="mi">4</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">],</span>
|
|
<span class="p">[</span><span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">],</span>
|
|
<span class="p">[</span><span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">],</span>
|
|
<span class="p">[</span><span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">]],</span> <span class="n">dtype</span><span class="o">=</span><span class="n">float32</span><span class="p">)</span>
|
|
</pre></div>
|
|
</div>
|
|
<p>To control the scatter location on an additional axis, add another index array to <code class="docutils literal notranslate"><span class="pre">indices</span></code> and another axis to <code class="docutils literal notranslate"><span class="pre">axes</span></code>:</p>
|
|
<div class="highlight-python notranslate"><div class="highlight"><pre><span></span><span class="n">auto</span> <span class="ow">in</span> <span class="o">=</span> <span class="n">zeros</span><span class="p">({</span><span class="mi">4</span><span class="p">,</span> <span class="mi">4</span><span class="p">},</span> <span class="n">float32</span><span class="p">);</span>
|
|
<span class="n">auto</span> <span class="n">indices</span> <span class="o">=</span> <span class="n">std</span><span class="p">::</span><span class="n">vector</span><span class="p">{</span><span class="n">array</span><span class="p">({</span><span class="mi">2</span><span class="p">,</span> <span class="mi">0</span><span class="p">}),</span> <span class="n">array</span><span class="p">({</span><span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">})};</span>
|
|
<span class="n">auto</span> <span class="n">updates</span> <span class="o">=</span> <span class="n">reshape</span><span class="p">(</span><span class="n">arange</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span class="mi">5</span><span class="p">,</span> <span class="n">float32</span><span class="p">),</span> <span class="p">{</span><span class="mi">2</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">});</span>
|
|
<span class="n">std</span><span class="p">::</span><span class="n">vector</span><span class="o"><</span><span class="nb">int</span><span class="o">></span> <span class="n">axes</span><span class="p">{</span><span class="mi">0</span><span class="p">,</span> <span class="mi">1</span><span class="p">};</span>
|
|
|
|
<span class="n">auto</span> <span class="n">out</span> <span class="o">=</span> <span class="n">scatter</span><span class="p">(</span><span class="ow">in</span><span class="p">,</span> <span class="n">indices</span><span class="p">,</span> <span class="n">updates</span><span class="p">,</span> <span class="n">axes</span><span class="p">);</span>
|
|
</pre></div>
|
|
</div>
|
|
<p>will produce:</p>
|
|
<div class="highlight-python notranslate"><div class="highlight"><pre><span></span><span class="n">array</span><span class="p">([[</span><span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span> <span class="mi">3</span><span class="p">,</span> <span class="mi">4</span><span class="p">],</span>
|
|
<span class="p">[</span><span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">],</span>
|
|
<span class="p">[</span><span class="mi">0</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">,</span> <span class="mi">0</span><span class="p">],</span>
|
|
<span class="p">[</span><span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">]],</span> <span class="n">dtype</span><span class="o">=</span><span class="n">float32</span><span class="p">)</span>
|
|
</pre></div>
|
|
</div>
|
|
<p>Items in indices are broadcasted together. This means:</p>
|
|
<div class="highlight-python notranslate"><div class="highlight"><pre><span></span><span class="n">auto</span> <span class="n">indices</span> <span class="o">=</span> <span class="n">std</span><span class="p">::</span><span class="n">vector</span><span class="p">{</span><span class="n">array</span><span class="p">({</span><span class="mi">2</span><span class="p">,</span> <span class="mi">0</span><span class="p">}),</span> <span class="n">array</span><span class="p">({</span><span class="mi">1</span><span class="p">})};</span>
|
|
</pre></div>
|
|
</div>
|
|
<p>is equivalent to:</p>
|
|
<div class="highlight-python notranslate"><div class="highlight"><pre><span></span><span class="n">auto</span> <span class="n">indices</span> <span class="o">=</span> <span class="n">std</span><span class="p">::</span><span class="n">vector</span><span class="p">{</span><span class="n">array</span><span class="p">({</span><span class="mi">2</span><span class="p">,</span> <span class="mi">0</span><span class="p">}),</span> <span class="n">array</span><span class="p">({</span><span class="mi">1</span><span class="p">,</span> <span class="mi">1</span><span class="p">})};</span>
|
|
</pre></div>
|
|
</div>
|
|
<p>Note, <code class="docutils literal notranslate"><span class="pre">scatter</span></code> does not perform bounds checking on the indices and updates. Out-of-bounds accesses on <code class="docutils literal notranslate"><span class="pre">a</span></code> are undefined and typically result in unintended or invalid memory writes. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv37scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice"></span><span id="_CPPv27scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice"></span><span id="scatter__arrayCR.arrayCR.arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1gac2c2b379a3ce959dbe1c4a68f112edfe"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">scatter</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">indices</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">updates</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice">
|
|
<span id="_CPPv311scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"></span><span id="_CPPv211scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"></span><span id="scatter_add__arrayCR.std::vector:array:CR.arrayCR.std::vector:i:CR.StreamOrDevice"></span><span class="target" id="group__ops_1gacd14c2b5cfebf343fc2d672722f8d174"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">scatter_add</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">indices</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">updates</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Scatter and add updates to given indices. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv311scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice"></span><span id="_CPPv211scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice"></span><span id="scatter_add__arrayCR.arrayCR.arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1gac13318518e5703f1273c5366eb523a5a"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">scatter_add</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">indices</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">updates</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice">
|
|
<span id="_CPPv312scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"></span><span id="_CPPv212scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"></span><span id="scatter_prod__arrayCR.std::vector:array:CR.arrayCR.std::vector:i:CR.StreamOrDevice"></span><span class="target" id="group__ops_1ga3708b5bcb61e2c63d213c4ce6ad0ffc0"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">scatter_prod</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">indices</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">updates</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Scatter and prod updates to given indices. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv312scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice"></span><span id="_CPPv212scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice"></span><span id="scatter_prod__arrayCR.arrayCR.arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1gaf83c53c453faa9083ba27e4b97539339"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">scatter_prod</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">indices</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">updates</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice">
|
|
<span id="_CPPv311scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"></span><span id="_CPPv211scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"></span><span id="scatter_max__arrayCR.std::vector:array:CR.arrayCR.std::vector:i:CR.StreamOrDevice"></span><span class="target" id="group__ops_1ga05881a4157cd113c9392d168a79e6673"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">scatter_max</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">indices</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">updates</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Scatter and max updates to given linear indices. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv311scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice"></span><span id="_CPPv211scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice"></span><span id="scatter_max__arrayCR.arrayCR.arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga9adda5f9202bb3486e4d9e1114e3a56f"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">scatter_max</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">indices</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">updates</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice">
|
|
<span id="_CPPv311scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"></span><span id="_CPPv211scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"></span><span id="scatter_min__arrayCR.std::vector:array:CR.arrayCR.std::vector:i:CR.StreamOrDevice"></span><span class="target" id="group__ops_1ga0ca16b7579dfc899f3f7fd40245ba7c5"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">scatter_min</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">indices</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">updates</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Scatter and min updates to given linear indices. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv311scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice"></span><span id="_CPPv211scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice"></span><span id="scatter_min__arrayCR.arrayCR.arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga51fa762a997c243ca7a19e1ed3e83199"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">scatter_min</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">indices</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">updates</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44sqrtRK5array14StreamOrDevice">
|
|
<span id="_CPPv34sqrtRK5array14StreamOrDevice"></span><span id="_CPPv24sqrtRK5array14StreamOrDevice"></span><span id="sqrt__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga297f853b3d90ec8ae81263977ba2ddb1"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">sqrt</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44sqrtRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Square root the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45rsqrtRK5array14StreamOrDevice">
|
|
<span id="_CPPv35rsqrtRK5array14StreamOrDevice"></span><span id="_CPPv25rsqrtRK5array14StreamOrDevice"></span><span id="rsqrt__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga102f23aa0b0c3d3296a321c694617aa1"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">rsqrt</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45rsqrtRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Square root and reciprocal the elements of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice">
|
|
<span id="_CPPv37softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"></span><span id="_CPPv27softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"></span><span id="softmax__arrayCR.std::vector:i:CR.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga7e9bb08b43c8fd0444b7d3c9e09dc1c6"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">softmax</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">precise</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Softmax of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47softmaxRK5arrayb14StreamOrDevice">
|
|
<span id="_CPPv37softmaxRK5arrayb14StreamOrDevice"></span><span id="_CPPv27softmaxRK5arrayb14StreamOrDevice"></span><span id="softmax__arrayCR.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga1ae3614d07d873892a530d14c3857d0b"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">softmax</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">precise</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47softmaxRK5arrayb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Softmax of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47softmaxRK5arrayib14StreamOrDevice">
|
|
<span id="_CPPv37softmaxRK5arrayib14StreamOrDevice"></span><span id="_CPPv27softmaxRK5arrayib14StreamOrDevice"></span><span id="softmax__arrayCR.i.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga06f570d73716a24303e6de3aaba4457b"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">softmax</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">precise</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47softmaxRK5arrayib14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Softmax of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45powerRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv35powerRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv25powerRK5arrayRK5array14StreamOrDevice"></span><span id="power__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga7972058715c26559dff9c9ae2a3ef76d"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">power</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45powerRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Raise elements of a to the power of b element-wise. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46cumsumRK5arrayibb14StreamOrDevice">
|
|
<span id="_CPPv36cumsumRK5arrayibb14StreamOrDevice"></span><span id="_CPPv26cumsumRK5arrayibb14StreamOrDevice"></span><span id="cumsum__arrayCR.i.b.b.StreamOrDevice"></span><span class="target" id="group__ops_1gaddc825a5c173e195ab0fda83ad630420"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">cumsum</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">reverse</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">inclusive</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">true</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46cumsumRK5arrayibb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Cumulative sum of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47cumprodRK5arrayibb14StreamOrDevice">
|
|
<span id="_CPPv37cumprodRK5arrayibb14StreamOrDevice"></span><span id="_CPPv27cumprodRK5arrayibb14StreamOrDevice"></span><span id="cumprod__arrayCR.i.b.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga0d71dfbc14ef3ed564b0c5ee26af680f"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">cumprod</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">reverse</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">inclusive</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">true</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47cumprodRK5arrayibb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Cumulative product of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46cummaxRK5arrayibb14StreamOrDevice">
|
|
<span id="_CPPv36cummaxRK5arrayibb14StreamOrDevice"></span><span id="_CPPv26cummaxRK5arrayibb14StreamOrDevice"></span><span id="cummax__arrayCR.i.b.b.StreamOrDevice"></span><span class="target" id="group__ops_1gaee37cac8476e8f8d666bcded5bc59143"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">cummax</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">reverse</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">inclusive</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">true</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46cummaxRK5arrayibb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Cumulative max of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46cumminRK5arrayibb14StreamOrDevice">
|
|
<span id="_CPPv36cumminRK5arrayibb14StreamOrDevice"></span><span id="_CPPv26cumminRK5arrayibb14StreamOrDevice"></span><span id="cummin__arrayCR.i.b.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga19c1bf6929fe8d66b9cd408946aea6a8"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">cummin</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">reverse</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">inclusive</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">true</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46cumminRK5arrayibb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Cumulative min of an array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice">
|
|
<span id="_CPPv312conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice"></span><span id="_CPPv212conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice"></span><span id="conv_general__array.array.std::vector:i:.std::vector:i:.std::vector:i:.std::vector:i:.std::vector:i:.i.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga2236e5dfc7e52e28abf6c21675d0a51e"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">conv_general</span></span></span><span class="sig-paren">(</span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">input</span></span>, <span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">weight</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">stride</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">padding_lo</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">padding_hi</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">kernel_dilation</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">input_dilation</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">groups</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">flip</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>General convolution with a filter. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice">
|
|
<span id="_CPPv312conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice"></span><span id="_CPPv212conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice"></span><span id="conv_general__arrayCR.arrayCR.std::vector:i:.std::vector:i:.std::vector:i:.std::vector:i:.i.b.StreamOrDevice"></span><span class="target" id="group__ops_1gab59f89942cd1efaadffe9e8762e3c99d"></span><span class="k"><span class="pre">inline</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">conv_general</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">input</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">weight</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">stride</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">padding</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">kernel_dilation</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">input_dilation</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">groups</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">flip</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>General convolution with a filter. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice">
|
|
<span id="_CPPv36conv1dRK5arrayRK5arrayiiii14StreamOrDevice"></span><span id="_CPPv26conv1dRK5arrayRK5arrayiiii14StreamOrDevice"></span><span id="conv1d__arrayCR.arrayCR.i.i.i.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga30d47e08093c03a3676f235f9f559411"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">conv1d</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">input</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">weight</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">stride</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">padding</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">dilation</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">groups</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>1D convolution with a filter </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice">
|
|
<span id="_CPPv36conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice"></span><span id="_CPPv26conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice"></span><span id="conv2d__arrayCR.arrayCR.std::pair:i.i:CR.std::pair:i.i:CR.std::pair:i.i:CR.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga73b02833229678786e7f302d458d5a83"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">conv2d</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">input</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">weight</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">pair</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">stride</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="m"><span class="pre">1</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span><span class="p"><span class="pre">}</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">pair</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">padding</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="m"><span class="pre">0</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span><span class="p"><span class="pre">}</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">pair</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">dilation</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="m"><span class="pre">1</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span><span class="p"><span class="pre">}</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">groups</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>2D convolution with a filter </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice">
|
|
<span id="_CPPv36conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice"></span><span id="_CPPv26conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice"></span><span id="conv3d__arrayCR.arrayCR.std::tuple:i.i.i:CR.std::tuple:i.i.i:CR.std::tuple:i.i.i:CR.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga6e9907d2f14dc4803e4306b3dbc4b3ca"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">conv3d</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">input</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">weight</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">tuple</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">stride</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="m"><span class="pre">1</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span><span class="p"><span class="pre">}</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">tuple</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">padding</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="m"><span class="pre">0</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span><span class="p"><span class="pre">}</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">tuple</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">dilation</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="m"><span class="pre">1</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span><span class="p"><span class="pre">}</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">groups</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>3D convolution with a filter </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice">
|
|
<span id="_CPPv316conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice"></span><span id="_CPPv216conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice"></span><span id="conv_transpose1d__arrayCR.arrayCR.i.i.i.i.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga6e02c7cd0a5844a260b3c6e7d45e0811"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">conv_transpose1d</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">input</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">weight</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">stride</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">padding</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">dilation</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">output_padding</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">groups</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>1D transposed convolution with a filter </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice">
|
|
<span id="_CPPv316conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice"></span><span id="_CPPv216conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice"></span><span id="conv_transpose2d__arrayCR.arrayCR.std::pair:i.i:CR.std::pair:i.i:CR.std::pair:i.i:CR.std::pair:i.i:CR.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga3644f52609b78ba0b27548ee503cc34c"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">conv_transpose2d</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">input</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">weight</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">pair</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">stride</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="m"><span class="pre">1</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span><span class="p"><span class="pre">}</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">pair</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">padding</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="m"><span class="pre">0</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span><span class="p"><span class="pre">}</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">pair</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">dilation</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="m"><span class="pre">1</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span><span class="p"><span class="pre">}</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">pair</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">output_padding</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="m"><span class="pre">0</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span><span class="p"><span class="pre">}</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">groups</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>2D transposed convolution with a filter </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice">
|
|
<span id="_CPPv316conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice"></span><span id="_CPPv216conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice"></span><span id="conv_transpose3d__arrayCR.arrayCR.std::tuple:i.i.i:CR.std::tuple:i.i.i:CR.std::tuple:i.i.i:CR.std::tuple:i.i.i:CR.i.StreamOrDevice"></span><span class="target" id="group__ops_1gaaa0cb6d891287c5b1b8dc87b5c48dd17"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">conv_transpose3d</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">input</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">weight</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">tuple</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">stride</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="m"><span class="pre">1</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span><span class="p"><span class="pre">}</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">tuple</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">padding</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="m"><span class="pre">0</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span><span class="p"><span class="pre">}</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">tuple</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">dilation</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="m"><span class="pre">1</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span><span class="p"><span class="pre">}</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">tuple</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">output_padding</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="m"><span class="pre">0</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span><span class="p"><span class="pre">}</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">groups</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>3D transposed convolution with a filter </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice">
|
|
<span id="_CPPv316quantized_matmul5array5array5array5arraybii14StreamOrDevice"></span><span id="_CPPv216quantized_matmul5array5array5array5arraybii14StreamOrDevice"></span><span id="quantized_matmul__array.array.array.array.b.i.i.StreamOrDevice"></span><span class="target" id="group__ops_1gabfa4208fb1f9b1cdd0abc563b19175af"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">quantized_matmul</span></span></span><span class="sig-paren">(</span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">x</span></span>, <span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">w</span></span>, <span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">scales</span></span>, <span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">biases</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">transpose</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">true</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">group_size</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">64</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">bits</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">4</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Quantized matmul multiplies x with a quantized matrix w. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv48quantizeRK5arrayii14StreamOrDevice">
|
|
<span id="_CPPv38quantizeRK5arrayii14StreamOrDevice"></span><span id="_CPPv28quantizeRK5arrayii14StreamOrDevice"></span><span id="quantize__arrayCR.i.i.StreamOrDevice"></span><span class="target" id="group__ops_1gab43cc28690da7cdd43b43065adbd31da"></span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">tuple</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">quantize</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">w</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">group_size</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">64</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">bits</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">4</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv48quantizeRK5arrayii14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Quantize a matrix along its last axis. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice">
|
|
<span id="_CPPv310dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice"></span><span id="_CPPv210dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice"></span><span id="dequantize__arrayCR.arrayCR.arrayCR.i.i.StreamOrDevice"></span><span class="target" id="group__ops_1gabff758a5c1ce32ad7e8b78aba0164077"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">dequantize</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">w</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">scales</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">biases</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">group_size</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">64</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">bits</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">4</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Dequantize a matrix produced by <a class="reference internal" href="#group__ops_1gab43cc28690da7cdd43b43065adbd31da"><span class="std std-ref">quantize()</span></a></p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbiib14StreamOrDevice">
|
|
<span id="_CPPv310gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbiib14StreamOrDevice"></span><span id="_CPPv210gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbiib14StreamOrDevice"></span><span id="gather_qmm__arrayCR.arrayCR.arrayCR.arrayCR.std::optional:array:.std::optional:array:.b.i.i.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga8a358d1d4721a5c3f2b84d42dd42f898"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">gather_qmm</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">x</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">w</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">scales</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">biases</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">optional</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">lhs_indices</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">nullopt</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">optional</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">rhs_indices</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">nullopt</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">transpose</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">true</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">group_size</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">64</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">bits</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">4</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">sorted_indices</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbiib14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Compute matrix products with matrix-level gather. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice">
|
|
<span id="_CPPv39tensordotRK5arrayRK5arrayKi14StreamOrDevice"></span><span id="_CPPv29tensordotRK5arrayRK5arrayKi14StreamOrDevice"></span><span id="tensordot__arrayCR.arrayCR.iC.StreamOrDevice"></span><span class="target" id="group__ops_1gaf5c9735f4690327e1500e04e728fae70"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">tensordot</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">2</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Returns a contraction of a and b over multiple dimensions. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice">
|
|
<span id="_CPPv39tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice"></span><span id="_CPPv29tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice"></span><span id="tensordot__arrayCR.arrayCR.std::vector:i:CR.std::vector:i:CR.StreamOrDevice"></span><span class="target" id="group__ops_1gad7fe00b566f89d607639c1a497cabbc6"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">tensordot</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes_a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes_b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45outerRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv35outerRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv25outerRK5arrayRK5array14StreamOrDevice"></span><span id="outer__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga866af24e10db2797e1c5a5986dbf6c0d"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">outer</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45outerRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Compute the outer product of two vectors. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45innerRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv35innerRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv25innerRK5arrayRK5array14StreamOrDevice"></span><span id="inner__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga654fec16a9746b390916697a2ab2546e"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">inner</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45innerRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Compute the inner product of two vectors. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice">
|
|
<span id="_CPPv35addmm5array5array5arrayRKfRKf14StreamOrDevice"></span><span id="_CPPv25addmm5array5array5arrayRKfRKf14StreamOrDevice"></span><span id="addmm__array.array.array.floatCR.floatCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga82a53e083205a965387b3c3e2463244a"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">addmm</span></span></span><span class="sig-paren">(</span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">c</span></span>, <span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">b</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="kt"><span class="pre">float</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">alpha</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">1.f</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="kt"><span class="pre">float</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">beta</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">1.f</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Compute D = beta * C + alpha * (A @ B) </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice">
|
|
<span id="_CPPv315block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice"></span><span id="_CPPv215block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice"></span><span id="block_masked_mm__array.array.i.std::optional:array:.std::optional:array:.std::optional:array:.StreamOrDevice"></span><span class="target" id="group__ops_1ga6b76c8ea46b19e6866af155fa5910be6"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">block_masked_mm</span></span></span><span class="sig-paren">(</span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">b</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">block_size</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">optional</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">mask_out</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">nullopt</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">optional</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">mask_lhs</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">nullopt</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">optional</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">mask_rhs</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">nullopt</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Compute matrix product with block masking. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice">
|
|
<span id="_CPPv39gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice"></span><span id="_CPPv29gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice"></span><span id="gather_mm__array.array.std::optional:array:.std::optional:array:.b.StreamOrDevice"></span><span class="target" id="group__ops_1gac610de1eac3aa4fdc577875108f7b59c"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">gather_mm</span></span></span><span class="sig-paren">(</span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">b</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">optional</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">lhs_indices</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">nullopt</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">optional</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">rhs_indices</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">nullopt</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">sorted_indices</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Compute matrix product with matrix-level gather. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv48diagonalRK5arrayiii14StreamOrDevice">
|
|
<span id="_CPPv38diagonalRK5arrayiii14StreamOrDevice"></span><span id="_CPPv28diagonalRK5arrayiii14StreamOrDevice"></span><span id="diagonal__arrayCR.i.i.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga9236b085a88ead3128ed8079d009cac6"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">diagonal</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">offset</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis1</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis2</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">1</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv48diagonalRK5arrayiii14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Extract a diagonal or construct a diagonal array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44diagRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv34diagRK5arrayi14StreamOrDevice"></span><span id="_CPPv24diagRK5arrayi14StreamOrDevice"></span><span id="diag__arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga11af511875640e1fa88e0ca87e199344"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">diag</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">k</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="m"><span class="pre">0</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44diagRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Extract diagonal from a 2d array or create a diagonal matrix. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice">
|
|
<span id="_CPPv35traceRK5arrayiii5Dtype14StreamOrDevice"></span><span id="_CPPv25traceRK5arrayiii5Dtype14StreamOrDevice"></span><span id="trace__arrayCR.i.i.i.Dtype.StreamOrDevice"></span><span class="target" id="group__ops_1gabf786129c7660ed8d5acb5499bc6fefd"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">trace</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">offset</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis1</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis2</span></span>, <span class="n"><span class="pre">Dtype</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">dtype</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Return the sum along a specified diagonal in the given array. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45traceRK5arrayiii14StreamOrDevice">
|
|
<span id="_CPPv35traceRK5arrayiii14StreamOrDevice"></span><span id="_CPPv25traceRK5arrayiii14StreamOrDevice"></span><span id="trace__arrayCR.i.i.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga5ed43c2dbf7d6cbddbaa2fd682deaafd"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">trace</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">offset</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis1</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis2</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45traceRK5arrayiii14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv45traceRK5array14StreamOrDevice">
|
|
<span id="_CPPv35traceRK5array14StreamOrDevice"></span><span id="_CPPv25traceRK5array14StreamOrDevice"></span><span id="trace__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gaf25c00108feaafaa6350a4434cb0062e"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">trace</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv45traceRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE">
|
|
<span id="_CPPv37dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE"></span><span id="_CPPv27dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE"></span><span id="depends__std::vector:array:CR.std::vector:array:CR"></span><span class="target" id="group__ops_1gac4a51a68fbe1725436b026d2fbb95759"></span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">depends</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">inputs</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">dependencies</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Implements the identity function but allows injecting dependencies to other arrays. </p>
|
|
<p>This ensures that these other arrays will have been computed when the outputs of this function are computed. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv410atleast_1dRK5array14StreamOrDevice">
|
|
<span id="_CPPv310atleast_1dRK5array14StreamOrDevice"></span><span id="_CPPv210atleast_1dRK5array14StreamOrDevice"></span><span id="atleast_1d__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gaba4d25e7a2bf87ba4feb7837ec7fa396"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">atleast_1d</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv410atleast_1dRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>convert an array to an atleast ndim array </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice">
|
|
<span id="_CPPv310atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice"></span><span id="_CPPv210atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice"></span><span id="atleast_1d__std::vector:array:CR.StreamOrDevice"></span><span class="target" id="group__ops_1ga08ca172ce80157c916c89dd0b45b95c5"></span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">atleast_1d</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv410atleast_2dRK5array14StreamOrDevice">
|
|
<span id="_CPPv310atleast_2dRK5array14StreamOrDevice"></span><span id="_CPPv210atleast_2dRK5array14StreamOrDevice"></span><span id="atleast_2d__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gaeeb7f5bb88aa32a3ac2be1f39c5f8087"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">atleast_2d</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv410atleast_2dRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice">
|
|
<span id="_CPPv310atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice"></span><span id="_CPPv210atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice"></span><span id="atleast_2d__std::vector:array:CR.StreamOrDevice"></span><span class="target" id="group__ops_1ga9950299a80c2562f13448758f856d1f5"></span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">atleast_2d</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv410atleast_3dRK5array14StreamOrDevice">
|
|
<span id="_CPPv310atleast_3dRK5array14StreamOrDevice"></span><span id="_CPPv210atleast_3dRK5array14StreamOrDevice"></span><span id="atleast_3d__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga4afd919601e67782ff964465919956a0"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">atleast_3d</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv410atleast_3dRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice">
|
|
<span id="_CPPv310atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice"></span><span id="_CPPv210atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice"></span><span id="atleast_3d__std::vector:array:CR.StreamOrDevice"></span><span class="target" id="group__ops_1gaffdf742ad79440a60dda40062a8074fe"></span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">atleast_3d</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="n"><span class="pre">array</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice">
|
|
<span id="_CPPv318number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice"></span><span id="_CPPv218number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice"></span><span id="number_of_elements__arrayCR.std::vector:i:.b.Dtype.StreamOrDevice"></span><span class="target" id="group__ops_1ga6d5f5f72362488b956cdc4615ef6c636"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">number_of_elements</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">inverted</span></span>, <span class="n"><span class="pre">Dtype</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">dtype</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="n"><span class="pre">int32</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Extract the number of elements along some axes as a scalar array. </p>
|
|
<p>Used to allow shape dependent shapeless compilation (pun intended). </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv49conjugateRK5array14StreamOrDevice">
|
|
<span id="_CPPv39conjugateRK5array14StreamOrDevice"></span><span id="_CPPv29conjugateRK5array14StreamOrDevice"></span><span id="conjugate__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga5b596906bf8cdc8d97ed6ddc9aeb4c23"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">conjugate</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv49conjugateRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv311bitwise_andRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv211bitwise_andRK5arrayRK5array14StreamOrDevice"></span><span id="bitwise_and__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga752fd2707dabb05d0308ba3d55346ada"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">bitwise_and</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Bitwise and. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4anRK5arrayRK5array">
|
|
<span id="_CPPv3anRK5arrayRK5array"></span><span id="_CPPv2anRK5arrayRK5array"></span><span id="and-operator__arrayCR.arrayCR"></span><span class="target" id="group__ops_1gaf0d232de4cbfffda1e2c838f8afdf6ff"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">&</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4anRK5arrayRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv310bitwise_orRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv210bitwise_orRK5arrayRK5array14StreamOrDevice"></span><span id="bitwise_or__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga8af4f22c08c11c4ffab7e3d45e0f3cd6"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">bitwise_or</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Bitwise inclusive or. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4orRK5arrayRK5array">
|
|
<span id="_CPPv3orRK5arrayRK5array"></span><span id="_CPPv2orRK5arrayRK5array"></span><span id="or-operator__arrayCR.arrayCR"></span><span class="target" id="group__ops_1ga52392a2a98f09a80da8d338c4908bd02"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">|</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4orRK5arrayRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv311bitwise_xorRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv211bitwise_xorRK5arrayRK5array14StreamOrDevice"></span><span id="bitwise_xor__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga3188638fba3a60e264baf69956a1e08b"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">bitwise_xor</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Bitwise exclusive or. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4eoRK5arrayRK5array">
|
|
<span id="_CPPv3eoRK5arrayRK5array"></span><span id="_CPPv2eoRK5arrayRK5array"></span><span id="xor-operator__arrayCR.arrayCR"></span><span class="target" id="group__ops_1gac3a6fe18694e84b3d63458e9553ac181"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">^</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4eoRK5arrayRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv310left_shiftRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv210left_shiftRK5arrayRK5array14StreamOrDevice"></span><span id="left_shift__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga89682bf78491761e062d4ee7bef0c829"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">left_shift</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Shift bits to the left. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4lsRK5arrayRK5array">
|
|
<span id="_CPPv3lsRK5arrayRK5array"></span><span id="_CPPv2lsRK5arrayRK5array"></span><span id="lshift-operator__arrayCR.arrayCR"></span><span class="target" id="group__ops_1gad656c30f9fd7d9467e405657b325aa7e"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre"><<</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4lsRK5arrayRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice">
|
|
<span id="_CPPv311right_shiftRK5arrayRK5array14StreamOrDevice"></span><span id="_CPPv211right_shiftRK5arrayRK5array14StreamOrDevice"></span><span id="right_shift__arrayCR.arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gafa376ad57d38ba87378f0272dc379b23"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">right_shift</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Shift bits to the right. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4rsRK5arrayRK5array">
|
|
<span id="_CPPv3rsRK5arrayRK5array"></span><span id="_CPPv2rsRK5arrayRK5array"></span><span id="rshift-operator__arrayCR.arrayCR"></span><span class="target" id="group__ops_1ga498b61f7e8f056ae00297fa0dc17303a"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">>></span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">b</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4rsRK5arrayRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv414bitwise_invertRK5array14StreamOrDevice">
|
|
<span id="_CPPv314bitwise_invertRK5array14StreamOrDevice"></span><span id="_CPPv214bitwise_invertRK5array14StreamOrDevice"></span><span id="bitwise_invert__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gaf1182ae7c049fbc9ee190f3e0fffbf83"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">bitwise_invert</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv414bitwise_invertRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Invert the bits. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv4coRK5array">
|
|
<span id="_CPPv3coRK5array"></span><span id="_CPPv2coRK5array"></span><span id="inv-operator__arrayCR"></span><span class="target" id="group__ops_1ga849365a62878579a33b3d3ad09bbc7be"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="k"><span class="pre">operator</span></span><span class="o"><span class="pre">~</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv4coRK5array" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice">
|
|
<span id="_CPPv34viewRK5arrayRK5Dtype14StreamOrDevice"></span><span id="_CPPv24viewRK5arrayRK5Dtype14StreamOrDevice"></span><span id="view__arrayCR.DtypeCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga3602aa91b7b124a0b41ec1b2137a1b02"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">view</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">Dtype</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">dtype</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44rollRK5arrayi14StreamOrDevice">
|
|
<span id="_CPPv34rollRK5arrayi14StreamOrDevice"></span><span id="_CPPv24rollRK5arrayi14StreamOrDevice"></span><span id="roll__arrayCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1gac40e48c69f9c715a767912c30836e75c"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">roll</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">shift</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44rollRK5arrayi14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd><p>Roll elements along an axis and introduce them on the other side. </p>
|
|
</dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44rollRK5arrayRK5Shape14StreamOrDevice">
|
|
<span id="_CPPv34rollRK5arrayRK5Shape14StreamOrDevice"></span><span id="_CPPv24rollRK5arrayRK5Shape14StreamOrDevice"></span><span id="roll__arrayCR.ShapeCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga5011d1a5735c64e5b91afa56c7e2cc02"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">roll</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">shift</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44rollRK5arrayRK5Shape14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44rollRK5arrayii14StreamOrDevice">
|
|
<span id="_CPPv34rollRK5arrayii14StreamOrDevice"></span><span id="_CPPv24rollRK5arrayii14StreamOrDevice"></span><span id="roll__arrayCR.i.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga8694ec137165752cb6d8a36a6b7c3436"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">roll</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">shift</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44rollRK5arrayii14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice">
|
|
<span id="_CPPv34rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice"></span><span id="_CPPv24rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice"></span><span id="roll__arrayCR.i.std::vector:i:CR.StreamOrDevice"></span><span class="target" id="group__ops_1ga665f502ecc96f1f4467556b784abf9ae"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">roll</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">shift</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice">
|
|
<span id="_CPPv34rollRK5arrayRK5Shapei14StreamOrDevice"></span><span id="_CPPv24rollRK5arrayRK5Shapei14StreamOrDevice"></span><span id="roll__arrayCR.ShapeCR.i.StreamOrDevice"></span><span class="target" id="group__ops_1ga79137f90bc44ac9e35f408c012701df9"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">roll</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">shift</span></span>, <span class="kt"><span class="pre">int</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">axis</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice">
|
|
<span id="_CPPv34rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice"></span><span id="_CPPv24rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice"></span><span id="roll__arrayCR.ShapeCR.std::vector:i:CR.StreamOrDevice"></span><span class="target" id="group__ops_1ga9d76930fb567a7d459ff96fb851abe36"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">roll</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">Shape</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">shift</span></span>, <span class="k"><span class="pre">const</span></span><span class="w"> </span><a class="reference internal" href="#_CPPv4StRK5arraybi14StreamOrDevice" title="std"><span class="n"><span class="pre">std</span></span></a><span class="p"><span class="pre">::</span></span><span class="n"><span class="pre">vector</span></span><span class="p"><span class="pre"><</span></span><span class="kt"><span class="pre">int</span></span><span class="p"><span class="pre">></span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">axes</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44realRK5array14StreamOrDevice">
|
|
<span id="_CPPv34realRK5array14StreamOrDevice"></span><span id="_CPPv24realRK5array14StreamOrDevice"></span><span id="real__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1gaf8913cabeb9fb193ba687aaeb2087764"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">real</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44realRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv44imagRK5array14StreamOrDevice">
|
|
<span id="_CPPv34imagRK5array14StreamOrDevice"></span><span id="_CPPv24imagRK5array14StreamOrDevice"></span><span id="imag__arrayCR.StreamOrDevice"></span><span class="target" id="group__ops_1ga7ff592a64d528f0cf4f3d098465da029"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">imag</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv44imagRK5array14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
<dl class="cpp function">
|
|
<dt class="sig sig-object cpp" id="_CPPv410contiguousRK5arrayb14StreamOrDevice">
|
|
<span id="_CPPv310contiguousRK5arrayb14StreamOrDevice"></span><span id="_CPPv210contiguousRK5arrayb14StreamOrDevice"></span><span id="contiguous__arrayCR.b.StreamOrDevice"></span><span class="target" id="group__ops_1ga8ab10aa6c41416d739791164a52b25d5"></span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">contiguous</span></span></span><span class="sig-paren">(</span><span class="k"><span class="pre">const</span></span><span class="w"> </span><span class="n"><span class="pre">array</span></span><span class="w"> </span><span class="p"><span class="pre">&</span></span><span class="n sig-param"><span class="pre">a</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">allow_col_major</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="k"><span class="pre">false</span></span>, <span class="n"><span class="pre">StreamOrDevice</span></span><span class="w"> </span><span class="n sig-param"><span class="pre">s</span></span><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="p"><span class="pre">{</span></span><span class="p"><span class="pre">}</span></span><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv410contiguousRK5arrayb14StreamOrDevice" title="Link to this definition">#</a><br /></dt>
|
|
<dd></dd></dl>
|
|
|
|
</section>
|
|
|
|
|
|
</article>
|
|
|
|
|
|
|
|
|
|
|
|
|
|
<footer class="prev-next-footer d-print-none">
|
|
|
|
<div class="prev-next-area">
|
|
<a class="left-prev"
|
|
href="../python/_autosummary/mlx.utils.tree_reduce.html"
|
|
title="previous page">
|
|
<i class="fa-solid fa-angle-left"></i>
|
|
<div class="prev-next-info">
|
|
<p class="prev-next-subtitle">previous</p>
|
|
<p class="prev-next-title">mlx.utils.tree_reduce</p>
|
|
</div>
|
|
</a>
|
|
<a class="right-next"
|
|
href="../dev/extensions.html"
|
|
title="next page">
|
|
<div class="prev-next-info">
|
|
<p class="prev-next-subtitle">next</p>
|
|
<p class="prev-next-title">Custom Extensions in MLX</p>
|
|
</div>
|
|
<i class="fa-solid fa-angle-right"></i>
|
|
</a>
|
|
</div>
|
|
</footer>
|
|
|
|
</div>
|
|
|
|
|
|
|
|
<div class="bd-sidebar-secondary bd-toc"><div class="sidebar-secondary-items sidebar-secondary__inner">
|
|
|
|
|
|
<div class="sidebar-secondary-item">
|
|
<div class="page-toc tocsection onthispage">
|
|
<i class="fa-solid fa-list"></i> Contents
|
|
</div>
|
|
<nav class="bd-toc-nav page-toc">
|
|
<ul class="visible nav section-nav flex-column">
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46arangeddd5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arange()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46arangeddd14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arange()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46arangedd5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arange()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46arangedd14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arange()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46aranged5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arange()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46aranged14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arange()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46arangeiii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arange()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46arangeii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arange()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46arangei14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arange()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48linspaceddi5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">linspace()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46astype5array5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">astype()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">as_strided()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44copy5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">copy()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44full5Shape5array5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">full()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44full5Shape5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">full()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">full()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0E4full5array5Shape1T14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">full()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45zerosRK5Shape5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">zeros()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45zerosRK5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">zeros()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410zeros_likeRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">zeros_like()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44onesRK5Shape5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">ones()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44onesRK5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">ones()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49ones_likeRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">ones_like()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43eyeiii5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">eye()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43eyei5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">eye()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43eyeii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">eye()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43eyeiii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">eye()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43eyei14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">eye()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48identityi5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">identity()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48identityi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">identity()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43triiii5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">tri()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43trii5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">tri()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44tril5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">tril()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44triu5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">triu()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47reshapeRK5array5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">reshape()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">unflatten()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47flattenRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">flatten()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47flattenRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">flatten()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">hadamard_transform()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">squeeze()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47squeezeRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">squeeze()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47squeezeRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">squeeze()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">expand_dims()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411expand_dimsRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">expand_dims()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">slice()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">slice()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">slice()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">slice()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">slice_update()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">slice_update()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">slice_update()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45splitRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">split()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45splitRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">split()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">split()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45splitRK5arrayRK5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">split()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">meshgrid()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">clip()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">concatenate()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">concatenate()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">stack()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">stack()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46repeatRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">repeat()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46repeatRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">repeat()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">tile()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">transpose()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">transpose()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48swapaxesRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">swapaxes()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48moveaxisRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">moveaxis()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">pad()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">pad()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">pad()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">pad()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49transposeRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">transpose()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">broadcast_to()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">broadcast_arrays()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45equalRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">equal()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4eqRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator==()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Eeq5array1TRK5array"><code class="docutils literal notranslate"><span class="pre">operator==()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Eeq5arrayRK5array1T"><code class="docutils literal notranslate"><span class="pre">operator==()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49not_equalRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">not_equal()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4neRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator!=()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Ene5array1TRK5array"><code class="docutils literal notranslate"><span class="pre">operator!=()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Ene5arrayRK5array1T"><code class="docutils literal notranslate"><span class="pre">operator!=()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47greaterRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">greater()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4gtRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator>()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Egt5array1TRK5array"><code class="docutils literal notranslate"><span class="pre">operator>()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Egt5arrayRK5array1T"><code class="docutils literal notranslate"><span class="pre">operator>()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">greater_equal()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4geRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator>=()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Ege5array1TRK5array"><code class="docutils literal notranslate"><span class="pre">operator>=()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Ege5arrayRK5array1T"><code class="docutils literal notranslate"><span class="pre">operator>=()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44lessRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">less()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4ltRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator<()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Elt5array1TRK5array"><code class="docutils literal notranslate"><span class="pre">operator<()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Elt5arrayRK5array1T"><code class="docutils literal notranslate"><span class="pre">operator<()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410less_equalRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">less_equal()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4leRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator<=()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Ele5array1TRK5array"><code class="docutils literal notranslate"><span class="pre">operator<=()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Ele5arrayRK5array1T"><code class="docutils literal notranslate"><span class="pre">operator<=()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">array_equal()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411array_equalRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">array_equal()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45isnanRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">isnan()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45isinfRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">isinf()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48isfiniteRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">isfinite()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48isposinfRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">isposinf()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48isneginfRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">isneginf()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">where()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">nan_to_num()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43allRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">all()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43allRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">all()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">allclose()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">isclose()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">all()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43allRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">all()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43anyRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">any()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43anyRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">any()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">any()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43anyRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">any()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43sumRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sum()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43sumRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sum()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sum()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43sumRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sum()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44meanRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">mean()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44meanRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">mean()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">mean()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44meanRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">mean()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43varRK5arraybi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">var()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43varRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">var()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">var()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43varRK5arrayibi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">var()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4StRK5arraybi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">std()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4StRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">std()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">std()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4StRK5arrayibi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">std()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44prodRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">prod()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44prodRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">prod()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">prod()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44prodRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">prod()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43maxRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">max()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43maxRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">max()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">max()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43maxRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">max()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43minRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">min()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43minRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">min()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">min()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43minRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">min()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46argminRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">argmin()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46argminRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">argmin()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46argminRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">argmin()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46argmaxRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">argmax()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46argmaxRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">argmax()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46argmaxRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">argmax()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44sortRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sort()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44sortRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sort()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47argsortRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">argsort()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47argsortRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">argsort()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49partitionRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">partition()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49partitionRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">partition()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412argpartitionRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">argpartition()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412argpartitionRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">argpartition()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44topkRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">topk()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44topkRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">topk()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412logcumsumexpRK5arrayibb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">logcumsumexp()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49logsumexpRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">logsumexp()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49logsumexpRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">logsumexp()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">logsumexp()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49logsumexpRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">logsumexp()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43absRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">abs()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48negativeRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">negative()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4miRK5array"><code class="docutils literal notranslate"><span class="pre">operator-()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44signRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sign()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411logical_notRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">logical_not()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411logical_andRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">logical_and()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4aaRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator&&()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410logical_orRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">logical_or()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4ooRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator||()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410reciprocalRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">reciprocal()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43addRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">add()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4plRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator+()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Epl5array1TRK5array"><code class="docutils literal notranslate"><span class="pre">operator+()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Epl5arrayRK5array1T"><code class="docutils literal notranslate"><span class="pre">operator+()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48subtractRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">subtract()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4miRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator-()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Emi5array1TRK5array"><code class="docutils literal notranslate"><span class="pre">operator-()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Emi5arrayRK5array1T"><code class="docutils literal notranslate"><span class="pre">operator-()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48multiplyRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">multiply()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4mlRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator*()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Eml5array1TRK5array"><code class="docutils literal notranslate"><span class="pre">operator*()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Eml5arrayRK5array1T"><code class="docutils literal notranslate"><span class="pre">operator*()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46divideRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">divide()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4dvRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator/()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4dvdRK5array"><code class="docutils literal notranslate"><span class="pre">operator/()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4dvRK5arrayd"><code class="docutils literal notranslate"><span class="pre">operator/()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46divmodRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">divmod()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">floor_divide()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49remainderRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">remainder()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4rmRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator%()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Erm5array1TRK5array"><code class="docutils literal notranslate"><span class="pre">operator%()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4I0Erm5arrayRK5array1T"><code class="docutils literal notranslate"><span class="pre">operator%()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47maximumRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">maximum()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47minimumRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">minimum()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45floorRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">floor()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44ceilRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">ceil()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46squareRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">square()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43expRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">exp()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43sinRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sin()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43cosRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">cos()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43tanRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">tan()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46arcsinRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arcsin()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46arccosRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arccos()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46arctanRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arctan()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47arctan2RK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arctan2()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44sinhRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sinh()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44coshRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">cosh()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44tanhRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">tanh()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47arcsinhRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arcsinh()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47arccoshRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arccosh()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47arctanhRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">arctanh()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47degreesRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">degrees()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47radiansRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">radians()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43logRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">log()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44log2RK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">log2()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45log10RK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">log10()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45log1pRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">log1p()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">logaddexp()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47sigmoidRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sigmoid()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv43erfRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">erf()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46erfinvRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">erfinv()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45expm1RK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">expm1()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv413stop_gradientRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">stop_gradient()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45roundRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">round()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45roundRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">round()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46matmulRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">matmul()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">gather()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">gather()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44kronRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">kron()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">take()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44takeRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">take()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44takeRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">take()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44takeRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">take()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">take_along_axis()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">put_along_axis()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter_add_axis()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter_add()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter_add()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter_prod()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter_prod()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter_max()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter_max()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter_min()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">scatter_min()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44sqrtRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">sqrt()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45rsqrtRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">rsqrt()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">softmax()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47softmaxRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">softmax()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47softmaxRK5arrayib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">softmax()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45powerRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">power()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46cumsumRK5arrayibb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">cumsum()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47cumprodRK5arrayibb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">cumprod()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46cummaxRK5arrayibb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">cummax()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46cumminRK5arrayibb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">cummin()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">conv_general()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">conv_general()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">conv1d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">conv2d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">conv3d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">conv_transpose1d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">conv_transpose2d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">conv_transpose3d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">quantized_matmul()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48quantizeRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">quantize()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">dequantize()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbiib14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">gather_qmm()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">tensordot()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">tensordot()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45outerRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">outer()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45innerRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">inner()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">addmm()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">block_masked_mm()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">gather_mm()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv48diagonalRK5arrayiii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">diagonal()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44diagRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">diag()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">trace()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45traceRK5arrayiii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">trace()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv45traceRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">trace()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE"><code class="docutils literal notranslate"><span class="pre">depends()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410atleast_1dRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">atleast_1d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">atleast_1d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410atleast_2dRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">atleast_2d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">atleast_2d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410atleast_3dRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">atleast_3d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">atleast_3d()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">number_of_elements()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv49conjugateRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">conjugate()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">bitwise_and()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4anRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator&()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">bitwise_or()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4orRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator|()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">bitwise_xor()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4eoRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator^()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">left_shift()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4lsRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator<<()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">right_shift()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4rsRK5arrayRK5array"><code class="docutils literal notranslate"><span class="pre">operator>>()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv414bitwise_invertRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">bitwise_invert()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv4coRK5array"><code class="docutils literal notranslate"><span class="pre">operator~()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">view()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44rollRK5arrayi14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">roll()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44rollRK5arrayRK5Shape14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">roll()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44rollRK5arrayii14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">roll()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">roll()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">roll()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">roll()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44realRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">real()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv44imagRK5array14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">imag()</span></code></a></li>
|
|
<li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#_CPPv410contiguousRK5arrayb14StreamOrDevice"><code class="docutils literal notranslate"><span class="pre">contiguous()</span></code></a></li>
|
|
</ul>
|
|
</nav></div>
|
|
|
|
</div></div>
|
|
|
|
|
|
</div>
|
|
<footer class="bd-footer-content">
|
|
|
|
<div class="bd-footer-content__inner container">
|
|
|
|
<div class="footer-item">
|
|
|
|
<p class="component-author">
|
|
By MLX Contributors
|
|
</p>
|
|
|
|
</div>
|
|
|
|
<div class="footer-item">
|
|
|
|
|
|
<p class="copyright">
|
|
|
|
© Copyright 2023, Apple.
|
|
<br/>
|
|
|
|
</p>
|
|
|
|
</div>
|
|
|
|
<div class="footer-item">
|
|
|
|
</div>
|
|
|
|
<div class="footer-item">
|
|
|
|
</div>
|
|
|
|
</div>
|
|
</footer>
|
|
|
|
|
|
</main>
|
|
</div>
|
|
</div>
|
|
|
|
<!-- Scripts loaded after <body> so the DOM is not blocked -->
|
|
<script src="../_static/scripts/bootstrap.js?digest=dfe6caa3a7d634c4db9b"></script>
|
|
<script src="../_static/scripts/pydata-sphinx-theme.js?digest=dfe6caa3a7d634c4db9b"></script>
|
|
|
|
<footer class="bd-footer">
|
|
</footer>
|
|
</body>
|
|
</html> |