擅长:python、mysql、java
<p>如果您对与NumPy中完全相同的类型(即<code>None</code>)感兴趣,那么<code>tf.newaxis</code>是<code>np.newaxis</code>的确切替代方法。</p>
<p>示例:</p>
<pre><code>In [71]: a1 = tf.constant([2,2], name="a1")
In [72]: a1
Out[72]: <tf.Tensor 'a1_5:0' shape=(2,) dtype=int32>
# add a new dimension
In [73]: a1_new = a1[tf.newaxis, :]
In [74]: a1_new
Out[74]: <tf.Tensor 'strided_slice_5:0' shape=(1, 2) dtype=int32>
# add one more dimension
In [75]: a1_new = a1[tf.newaxis, :, tf.newaxis]
In [76]: a1_new
Out[76]: <tf.Tensor 'strided_slice_6:0' shape=(1, 2, 1) dtype=int32>
</code></pre>
<p>这和你在纽比做的手术完全一样。只需在你想增加它的地方使用它。</p>