擅长:python、mysql、java
<p>你可以定义一个函数</p>
<pre><code>def broadcast_axis(data, ndims, axis):
newshape = [1] * ndims
newshape[axis] = -1
return data.reshape(*newshape)
</code></pre>
<p>像这样使用它</p>
<pre><code>vector = broadcast_axis(vector, matrix.ndim, 3)
</code></pre>