tensorflow - Conv2DCustomBackpropInputOp 真的只支持NHWC吗？

Question

问题是当我使用 TensorFlow 2.0.0 和 Python 3.7 来训练用 TensorFlow 1.0.0 和 Python 2.7 编写的 gan 时，我发现 Conv2DCustomBackpropInputOp 只支持 NHWC。是这种情况吗？

Answer 1

最有可能的是，您的输入数据的形状可能采用以下形式，NCHW (Batch Size, Number of Channels, Height of the Image and Width of the Image).
将其形状更改为形式，NHWC (Batch Size, Height of the Image, Width of the Image and Number of Channels)并与 data_format='channels_last'应该可以解决您的问题。