擅长:python、mysql、java
<p>一种方法是分配一个空数组</p>
<pre><code>>> out = np.empty((N, 5, 5, 2), dtype=int)
</code></pre>
<p>然后用广播,例如</p>
^{pr2}$
<p>或者</p>
<pre><code>>>> out[...] = np.moveaxis(np.indices((5, 5)), 0, 2)
</code></pre>
<p>或者</p>
<pre><code>>>> out[..., 0] = np.arange(5)[None, :, None]
>>> out[..., 1] = np.arange(5)[None, None, :]
</code></pre>