有时我们会碰到升维或降维的需求,比如现在有一个图像样本,形状是 [height, width, channels],我们需要把它输入到已经训练好的模型中做分类,而模型定义的输入变量是一个batch,即形状为 [batch_size, height, width, channels],这时就需要升维了.tensorflow提供了一个方便的升维函数:expand_dims,参数定义如下: tf.expand_dims(input, axis=None, name=None, dim=None) 参数…