ef86d2e73e7ce03c4184a04a336d96caf661269a,sonnet/python/modules/conv.py,Conv1DTranspose,_build,#Conv1DTranspose#Any#,1292

Before Change

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

Instances