张量流中的卷积2D



对于tensorflow中GAN生成器的(16100(输入噪声,我必须使用ConvTranspose2d,而没有密集层。有什么建议吗?

我在pytorch中看到了这是可能的,但在tensorflow中没有。

有关背景信息:将pytorch发生器转换为tensorflow

让我们假设,您的数据是x,它有一个形状(16100(批处理大小为"b"。因此,要输入大小(16100(,您必须要将您的输入重塑为convertranspose2d,需要4维输入(batch_size、高度、宽度、通道(。所以你将重塑为如下:

x = x.reshape((16,100,1))

其中最后一个轴是通道轴。

现在,当你在convertransposition2d中输入时,is将有参数将input_shape参数设置为(16100,1(并自动添加批次尺寸轴到输入,这将使其成为4维,现在形状为(b,16100,1(。

最新更新