擅长:python、mysql、java
<p>相应的命令是<code>tf.newaxis</code>(或者<code>None</code>,如在numpy中)。在tensorflow的文档中,它本身没有一个条目,但是在<a href="https://www.tensorflow.org/api_docs/python/tf/strided_slice" rel="noreferrer">^{<cd3>}</a>的doc页面中有简要介绍。</p>
<pre><code>x = tf.ones((10,10,10))
y = x[:, tf.newaxis] # or y = x [:, None]
print(y.shape)
# prints (10, 1, 10, 10)
</code></pre>
<p>使用<code>tf.expand_dims</code>也可以,但是,如上面链接所述</p>
<blockquote>
<p>Those interfaces are much more friendly, and highly recommended.</p>
</blockquote>