tensorflow中squeeze与expand_dims
tf.squeezez = tf.squeeze(y) #删除y中所有为1的维度;z = tf.squeeze(y,[3,5]) #删除[3,5]中所有为1的维度;tf.expand_dimstf中没有tf.unsqueeze,一般写tf.expand_dims()扩增维度;one_img = tf.expand_dims(one_img, 0) #第0维插入1维one_img = tf.expand_dims(one_img, -1) #-1表示最后一维...
复制链接