https://blog.csdn.net/qq_31780525/article/details/72280284

tf.expand_dims(tensor,axis=0) shape(tensor)为1时expand_dims 插入一个维度 就是插入数值1

当shape(tensor)>=2是  

expand_dims :就是在该维度的外面上增加一对中括号[]  

import tensorflow as tf

a=tf.constant(2,shape=(2,3))

sess=tf.InteractiveSession()

a.eval()  #a的从左到右 第二个中括号才是 维度0 里面的数字是维度1   
#array([[2, 2, 2],
#       [2, 2, 2]])

b=tf.expand_dims(a,axis=0)tf.expand_dims和tf.squeeze函数
b.eval()
#array([[[2, 2, 2],
  #      [2, 2, 2]]])

b1=tf.squeeze(b)  #删除尺寸为1的维度  
b1.eval()
array([[2, 2, 2],
       [2, 2, 2]]) 

c=tf.expand_dims(a,axis=1)
c.eval()

#array([[[2, 2, 2]],

  #     [[2, 2, 2]]])

c1=tf.squeeze(c)
c1.eval()
array([[2, 2, 2],
       [2, 2, 2]])
 

e=tf.expand_dims(a,axis=2)
e.eval()
    #array([[[2],
    #        [2],
    #        [2]],
    
    #       [[2],
    #        [2],
    #        [2]]])
 

相关文章:

  • 2021-11-03
  • 2022-12-23
  • 2021-09-12
  • 2021-12-31
  • 2021-11-19
  • 2022-12-23
  • 2022-02-03
猜你喜欢
  • 2022-02-16
  • 2021-11-13
  • 2022-12-23
  • 2021-06-12
  • 2021-09-30
相关资源
相似解决方案