Deploying to gh-pages from @ 3bc1070 🚀

Author: lucascolley

generated/array_api_extra.create_diagonal.html

@@ -276,15 +276,16 @@ <h1>array_api_extra.create_diagonal<a class="headerlink" href="#array-api-extra-
 <dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>
 <dd class="field-odd"><ul class="simple">
-<li><p><strong>x</strong> (<em>array</em>) – A 1-D array.</p></li>
+<li><p><strong>x</strong> (<em>array</em>) – An array having shape <code class="docutils literal notranslate"><span class="pre">(*batch_dims,</span> <span class="pre">k)</span></code>.</p></li>
 <li><p><strong>offset</strong> (<a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.13)"><em>int</em></a><em>, </em><em>optional</em>) – Offset from the leading diagonal (default is <code class="docutils literal notranslate"><span class="pre">0</span></code>).
 Use positive ints for diagonals above the leading diagonal,
 and negative ints for diagonals below the leading diagonal.</p></li>
 <li><p><strong>xp</strong> (<em>array_namespace</em><em>, </em><em>optional</em>) – The standard-compatible namespace for <cite>x</cite>. Default: infer.</p></li>
 </ul>
 </dd>
 <dt class="field-even">Returns<span class="colon">:</span></dt>
-<dd class="field-even"><p>A 2-D array with <cite>x</cite> on the diagonal (offset by <cite>offset</cite>).</p>
+<dd class="field-even"><p>An array having shape <code class="docutils literal notranslate"><span class="pre">(*batch_dims,</span> <span class="pre">k+abs(offset),</span> <span class="pre">k+abs(offset))</span></code> with <cite>x</cite>
+on the diagonal (offset by <cite>offset</cite>).</p>
 </dd>
 <dt class="field-odd">Return type<span class="colon">:</span></dt>
 <dd class="field-odd"><p>array</p>