擅长:python、mysql、java
<p>我想应该是<a href="https://www.tensorflow.org/api_docs/python/tf/expand_dims" rel="noreferrer">^{<cd1>}</a>-</p>
<pre><code>tf.expand_dims(a, 1) # Or tf.expand_dims(a, -1)
</code></pre>
<p>基本上,我们列出了要插入新轴的轴ID,并将后轴/维度向后推<em>。</p>
<p>在链接的文档中,这里有几个扩展维度的示例-</p>
<pre><code># 't' is a tensor of shape [2]
shape(expand_dims(t, 0)) ==> [1, 2]
shape(expand_dims(t, 1)) ==> [2, 1]
shape(expand_dims(t, -1)) ==> [2, 1]
# 't2' is a tensor of shape [2, 3, 5]
shape(expand_dims(t2, 0)) ==> [1, 2, 3, 5]
shape(expand_dims(t2, 2)) ==> [2, 3, 1, 5]
shape(expand_dims(t2, 3)) ==> [2, 3, 5, 1]
</code></pre>