47c71c0524c7804863f2f4bcffdef5a873d84a55,txtgen/modules/connectors/connectors.py,,_mlp_transform,#,18

Before Change

    flat_output = tf.split(fc_output, size_list, axis=1)
    for (i, shape) in enumerate(shape_list):
        flat_output[i] = tf.reshape(flat_output[i], tf.TensorShape(batch_size).concatenate(shape))
    output = nest.pack_sequence_as(structure=output_size, flat_sequence=flat_output)

    return output

After Change

    flat_output = tf.split(fc_output, size_list, axis=1)

    if isinstance(flat_output_size[0], tf.TensorShape):
        for (i, shape) in enumerate(flat_output_size):
            new_shape = tf.TensorShape(batch_size).concatenate(shape)
            flat_output[i] = tf.reshape(flat_output[i], new_shape)
    output = nest.pack_sequence_as(structure=output_size,
                                   flat_sequence=flat_output)

    return output

Instances