在Tensorflow中如何处理输入和输出的backprop必须具有相同的批处理大小

Question

我试图重复一个实验，这篇论文所涉及的复合函数梯度学习的生成对抗模型。在这个过程中，我遇到了一个我从未见过的错误。我不知道是什么原因造成这个问题的。我希望有人能帮我。非常感谢！

下面是我的编码。我想问题是自发电机组（索引，重用=True）'。

自我.zs定义如下。

self.sz = np.random.uniform(-1, 1, size=(self.sz_num, self.z_dim)).astype(np.float32) 

我只想从中抽取64块数据自我作为发电机的输入。

^{pr2}$

每次运行此代码时，我都会得到如下错误。

^{pr3}$

关于我的生成器的详细信息如下所示。

^{pr4}$

Answer 1

我知道我的代码有什么问题。由于我的粗心，我应该写这行如下。在

index_g = self.sz[i*self.batch_size:(i+1)*self.batch_size]

在这种情况下，我再也不会出错了。在