def compute_output_shape(self, input_shape):
if self.transpose:
shape = Conv2DTranspose.compute_output_shape(self, input_shape)
else:
shape = super(GConv2D, self).compute_output_shape(input_shape)
nto = shape[3]
if self.h_output == 'C4':
nto *= 4
elif self.h_output == 'D4':
nto *= 8
return (shape[0], shape[1], shape[2], nto)
