Tensorflow实现cnn中的维度问题

Posted 未曾悟道的佛

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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即可

以上是关于Tensorflow实现cnn中的维度问题的主要内容,如果未能解决你的问题,请参考以下文章