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).

将其形状更改为 form, NHWC (Batch Size, Height of the Image, Width of the Image and Number of Channels)and withdata_format='channels_last'应该可以解决您的问题。