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即可
相關文章
相關標籤/搜索