|
332 | 332 | </div> |
333 | 333 | <dl class="py class"> |
334 | 334 | <dt class="sig sig-object py" id="vortex_torch.abs.tensor.FORMAT"> |
335 | | -<span class="property"><span class="k"><span class="pre">class</span></span><span class="w"> </span></span><span class="sig-prename descclassname"><span class="pre">vortex_torch.abs.tensor.</span></span><span class="sig-name descname"><span class="pre">FORMAT</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="o"><span class="pre">*</span></span><span class="n"><span class="pre">values</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/vortex_torch/abs/tensor.html#FORMAT"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#vortex_torch.abs.tensor.FORMAT" title="Link to this definition">¶</a></dt> |
| 335 | +<span class="property"><span class="k"><span class="pre">class</span></span><span class="w"> </span></span><span class="sig-name descname"><span class="pre">FORMAT</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="o"><span class="pre">*</span></span><span class="n"><span class="pre">values</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/vortex_torch/abs/tensor.html#FORMAT"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#vortex_torch.abs.tensor.FORMAT" title="Link to this definition">¶</a></dt> |
336 | 336 | <dd><p>Bases: <a class="reference external" href="https://docs.python.org/3/library/enum.html#enum.Enum" title="(in Python v3.14)"><code class="xref py py-class docutils literal notranslate"><span class="pre">Enum</span></code></a></p> |
337 | 337 | <p>Tensor storage/layout format.</p> |
338 | 338 | <dl class="py attribute"> |
|
372 | 372 |
|
373 | 373 | <dl class="py class"> |
374 | 374 | <dt class="sig sig-object py" id="vortex_torch.abs.tensor.vTensor"> |
375 | | -<span class="property"><span class="k"><span class="pre">class</span></span><span class="w"> </span></span><span class="sig-prename descclassname"><span class="pre">vortex_torch.abs.tensor.</span></span><span class="sig-name descname"><span class="pre">vTensor</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">shape</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">()</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dtype</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">torch.bfloat16</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">device</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">_format</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">FORMAT.BATCHED</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">tensor_id</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">-1</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">padded_shape</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/vortex_torch/abs/tensor.html#vTensor"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#vortex_torch.abs.tensor.vTensor" title="Link to this definition">¶</a></dt> |
| 375 | +<span class="property"><span class="k"><span class="pre">class</span></span><span class="w"> </span></span><span class="sig-name descname"><span class="pre">vTensor</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">shape</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">()</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dtype</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">torch.bfloat16</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">device</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">_format</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">FORMAT.BATCHED</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">tensor_id</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">-1</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">padded_shape</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/vortex_torch/abs/tensor.html#vTensor"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#vortex_torch.abs.tensor.vTensor" title="Link to this definition">¶</a></dt> |
376 | 376 | <dd><p>Bases: <a class="reference external" href="https://docs.python.org/3/library/functions.html#object" title="(in Python v3.14)"><code class="xref py py-class docutils literal notranslate"><span class="pre">object</span></code></a></p> |
377 | 377 | <p>Pure-metadata virtual tensor.</p> |
378 | 378 | <p>Carries the descriptor fields used by the graph builder, the |
|
471 | 471 |
|
472 | 472 | <dl class="py function"> |
473 | 473 | <dt class="sig sig-object py" id="vortex_torch.abs.tensor.as_vtensor"> |
474 | | -<span class="sig-prename descclassname"><span class="pre">vortex_torch.abs.tensor.</span></span><span class="sig-name descname"><span class="pre">as_vtensor</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">x</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">_format</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">FORMAT.BATCHED</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">tensor_id</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">-1</span></span></em>, <em class="sig-param"><span class="keyword-only-separator o"><abbr title="Keyword-only parameters separator (PEP 3102)"><span class="pre">*</span></abbr></span></em>, <em class="sig-param"><span class="n"><span class="pre">shape</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dtype</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">device</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/vortex_torch/abs/tensor.html#as_vtensor"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#vortex_torch.abs.tensor.as_vtensor" title="Link to this definition">¶</a></dt> |
| 474 | +<span class="sig-name descname"><span class="pre">as_vtensor</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">x</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">_format</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">FORMAT.BATCHED</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">tensor_id</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">-1</span></span></em>, <em class="sig-param"><span class="keyword-only-separator o"><abbr title="Keyword-only parameters separator (PEP 3102)"><span class="pre">*</span></abbr></span></em>, <em class="sig-param"><span class="n"><span class="pre">shape</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dtype</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">device</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/vortex_torch/abs/tensor.html#as_vtensor"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#vortex_torch.abs.tensor.as_vtensor" title="Link to this definition">¶</a></dt> |
475 | 475 | <dd><p>Build a <a class="reference internal" href="#vortex_torch.abs.tensor.vTensor" title="vortex_torch.abs.tensor.vTensor"><code class="xref py py-class docutils literal notranslate"><span class="pre">vTensor</span></code></a>.</p> |
476 | 476 | <p>Three calling styles, all returning a fresh <code class="docutils literal notranslate"><span class="pre">vTensor</span></code> (or, for an |
477 | 477 | existing <code class="docutils literal notranslate"><span class="pre">vTensor</span></code>, the same object re-tagged):</p> |
|
0 commit comments