我正在编写一段代码,以便使用我编写的库使用cudnn在c++中运行TensorFlow模型。但是它在扁平层上给了我奇怪的结果。在我的模型中,我指定首先在每一层上使用通道。因此,假设前一个maxpool的层的输出是形状批次,通道,高宽,它将被输入到扁平层。我在这里指定我首先在扁平层中使用通道。但是扁平层在相反的方向上给我结果(通道最后)。 例如,请参见以下输入 inputs = np.arange(75.0).reshape((1,3,5,5))
k = 0;
for n in range(1):
for h in range(3):
for w in range(5)