import tensorflow as tf

            x = tf.matmul(
                x, weights)  # (batch_size * self._num_nodes, output_size)

            biases = tf.get_variable("biases", [output_size],
                                     dtype=dtype,
                                     initializer=tf.constant_initializer(
                                         bias_start, dtype=dtype))
            x = tf.nn.bias_add(x, biases)
        # Reshape res back to: (batch_size, num_node, state_dim)
        return tf.reshape(x, [batch_size, self._num_nodes, output_size])