|
14 | 14 | </script> |
15 | 15 |
|
16 | 16 | <meta name="viewport" content="width=device-width, initial-scale=1.0" /> |
17 | | - <title>dpnp.dpnp_array — Data Parallel Extension for NumPy 0.17.0dev0+86.g8cf8a75d6e8 documentation</title> |
| 17 | + <title>dpnp.dpnp_array — Data Parallel Extension for NumPy 0.17.0dev1+7.ga244bf51497 documentation</title> |
18 | 18 | <link rel="stylesheet" type="text/css" href="../../_static/pygments.css?v=fa44fd50" /> |
19 | 19 | <link rel="stylesheet" type="text/css" href="../../_static/css/theme.css?v=e59714d7" /> |
20 | 20 |
|
21 | 21 |
|
22 | 22 | <script src="../../_static/jquery.js?v=5d32c60e"></script> |
23 | 23 | <script src="../../_static/_sphinx_javascript_frameworks_compat.js?v=2cd50e6c"></script> |
24 | | - <script src="../../_static/documentation_options.js?v=ac11ea5c"></script> |
| 24 | + <script src="../../_static/documentation_options.js?v=33080d50"></script> |
25 | 25 | <script src="../../_static/doctools.js?v=9a2dae69"></script> |
26 | 26 | <script src="../../_static/sphinx_highlight.js?v=dc90522c"></script> |
27 | 27 | <script src="../../_static/js/theme.js"></script> |
@@ -110,6 +110,7 @@ <h1>Source code for dpnp.dpnp_array</h1><div class="highlight"><pre> |
110 | 110 | <span class="c1"># *****************************************************************************</span> |
111 | 111 |
|
112 | 112 | <span class="kn">import</span> <span class="nn">dpctl.tensor</span> <span class="k">as</span> <span class="nn">dpt</span> |
| 113 | +<span class="kn">from</span> <span class="nn">dpctl.tensor._numpy_helper</span> <span class="kn">import</span> <span class="n">AxisError</span> |
113 | 114 |
|
114 | 115 | <span class="kn">import</span> <span class="nn">dpnp</span> |
115 | 116 |
|
@@ -484,7 +485,15 @@ <h1>Source code for dpnp.dpnp_array</h1><div class="highlight"><pre> |
484 | 485 | <span class="k">else</span><span class="p">:</span> |
485 | 486 | <span class="n">axes</span> <span class="o">=</span> <span class="p">[(</span><span class="o">-</span><span class="mi">2</span><span class="p">,</span> <span class="o">-</span><span class="mi">1</span><span class="p">),</span> <span class="p">(</span><span class="o">-</span><span class="mi">2</span><span class="p">,</span> <span class="o">-</span><span class="mi">1</span><span class="p">),</span> <span class="p">(</span><span class="o">-</span><span class="mi">2</span><span class="p">,</span> <span class="o">-</span><span class="mi">1</span><span class="p">)]</span> |
486 | 487 |
|
487 | | - <span class="n">dpnp</span><span class="o">.</span><span class="n">matmul</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">other</span><span class="p">,</span> <span class="n">out</span><span class="o">=</span><span class="bp">self</span><span class="p">,</span> <span class="n">axes</span><span class="o">=</span><span class="n">axes</span><span class="p">)</span> |
| 488 | + <span class="k">try</span><span class="p">:</span> |
| 489 | + <span class="n">dpnp</span><span class="o">.</span><span class="n">matmul</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">other</span><span class="p">,</span> <span class="n">out</span><span class="o">=</span><span class="bp">self</span><span class="p">,</span> <span class="n">axes</span><span class="o">=</span><span class="n">axes</span><span class="p">)</span> |
| 490 | + <span class="k">except</span> <span class="n">AxisError</span><span class="p">:</span> |
| 491 | + <span class="c1"># AxisError should indicate that the axes argument didn't work out</span> |
| 492 | + <span class="c1"># which should mean the second operand not being 2 dimensional.</span> |
| 493 | + <span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span> |
| 494 | + <span class="s2">"inplace matrix multiplication requires the first operand to "</span> |
| 495 | + <span class="s2">"have at least one and the second at least two dimensions."</span> |
| 496 | + <span class="p">)</span> |
488 | 497 | <span class="k">return</span> <span class="bp">self</span> |
489 | 498 |
|
490 | 499 | <span class="k">def</span> <span class="fm">__imod__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">other</span><span class="p">):</span> |
|
0 commit comments