code: def embed(self, n_id):
      batch_size = tf.shape(n_id)[0]
      n_id = tf.reshape(n_id, [-1])
      emb = self.gnn(n_id)
      emb = tf.reshape(emb, [batch_size, -1, self.dim])
      return emb
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
  import tf_euler
  from tf_euler.python.utils.encoders import ShallowEncoder
  from tf_euler.python.mp_utils.base_gnn import BaseGNNNet
  from tf_euler.python.mp_utils.base_gae import BaseGraphAutoEncoder
label_desc: Use image node_id 2 for calling the GraphAutoEncoder obj's underlying member method code with example usage: obj.embed(n_id) and returns: emb