import tensorflow as tf

        variables = [p.free_state for p in self.params if not p.fixed]
        variables = utils.unique(variables)
        free_state = tf.concat(0, [tf.reshape(v, [-1]) for v in variables])

