"Number of input channels must be known at module build time")
input_channels = self._input_shape[2]
if self._input_shape[0] is None:
raise base.UnderspecifiedError(
"Batch size must be known at module build time")
batch_size = self._input_shape[0]
After Change
out_shape = (1, self.output_shape[0])
out_channels = (self.output_channels,)
out_shape_tuple = out_shape + out_channels
conv_output_shape = tf.convert_to_tensor(out_shape_tuple)
tf_out_shape = tf.concat([batch_size, conv_output_shape], 0)
// Add an extra dimension to the input - a height of 1.
inputs = tf.expand_dims(inputs, 1)
