tensorflow中squeeze与expend_dims

tf.squeeze

• z = tf.squeeze(y) #删除y中所有为1的维度；
• z = tf.squeeze(y，[3，5]) #删除[3，5]中所有为1的维度；

tf.expand_dims

• tf中没有tf.unsqueeze，一般写tf.expand_dims()扩增维度；
• one_img = tf.expand_dims(one_img, 0) #第0维插入1维
• one_img = tf.expand_dims(one_img, -1) #-1表示最后一维