使用TensorFlow实现cnn,其中tf.nn.conv2d(input_tensor…) input_tensor的格式要求是[batch, in_height, in_width, in_channels]
但是python中四维矩阵是[batch, in_channels,in_height, in_width]
使用transpose即可
使用TensorFlow实现cnn,其中tf.nn.conv2d(input_tensor…) input_tensor的格式要求是[batch, in_height, in_width, in_channels]
但是python中四维矩阵是[batch, in_channels,in_height, in_width]
使用transpose即可