Tensorflow实现cnn中的维度问题

使用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即可
相关文章
相关标签/搜索