import tensorflow as tf

        :param bias_start:
        :param scope:
        :return:
        """
        # Reshape input to (batch_size, num_nodes, input_dim)
        output_size = self._num_units
        batch_size = inputs.get_shape()[0].value
        inputs = tf.reshape(inputs, [batch_size, self._num_nodes, -1])
        input_size = inputs.get_shape()[2].value
        dtype = inputs.dtype

        x = inputs
        x0 = tf.transpose(x, perm=[1, 2,0])  # (num_nodes, total_arg_size, batch_size)
        x0 = tf.reshape(x0, shape=[self._num_nodes, input_size * batch_size])
