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中的维度问题的主要内容,如果未能解决你的问题,请参考以下文章
tensorflow框架学习 —— CNN卷积神经网络的实现
深度学习之卷积神经网络CNN及tensorflow代码实现示例
《神经网络与深度学习》 卷积神经网络CNN及tensorflow代码实现示例