tf.expand_dims()

tf.expand_dims()

 

例1

import tensorflow as tf
c = [[3, 3, 3, 6], [4, 4, 7, 5]]  # 维度是[2, 4]
d = tf.expand_dims(c, -2)  
e = tf.expand_dims(c, 0)
f = tf.expand_dims(c, -1)
with tf.Session() as sess:
    print(sess.run(d))
    print("-----------------------")
    print(sess.run(e))
    print("-----------------------")
    print(sess.run(f))

输出：

[[[3 3 3 6]]

 [[4 4 7 5]]]   # 维度是[2, 1, 4]
-----------------------
[[[3 3 3 6]
  [4 4 7 5]]]   # 维度是[1, 2, 4]

-----------------------
[[[3]
  [3]
  [3]
  [6]]

 [[4]
  [4]
  [7]
  [5]]]  # 维度是[2,4,1]