code: def to_sample(self, inputs):
      batch_size = tf.size(inputs)
      src = tf.expand_dims(inputs, -1)
      pos = tf_euler.sample_neighbor(
          inputs, self.edge_type, self.num_negs, self.max_id + 1
      )[0]
      negs = tf_euler.sample_node(
          batch_size * self.num_negs, self.node_type
      )
      negs = tf.reshape(negs, [batch_size, self.num_negs])
      return src, pos, negs
  
file_path: euler/tf_euler/python/mp_utils/base_gae.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
  import tf_euler
label_desc: Use image node_id 2 for calling the BaseGraphAutoEncoder obj's underlying member method code with example usage: obj.to_sample(inputs) and returns: src,  pos,  negs