tensorflow中squeeze与expand

tensorflow中squeeze与expand 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表示最后一维