code: def transfer_embedding(self, source, emb):
      if self.params["infer_type"] == "node_src":
          return source[:, 0], emb[0]
      elif self.params["infer_type"] == "edge":
          return source[:, :2], emb[1]
      elif self.params["infer_type"] == "node_dst":
          return source[:, 1], emb[2]
      else:
          raise ValueError("infer_type must be node_src/node_dst/edge.")
  
file_path: euler/euler_estimator/python/edge_estimator.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
  import numpy
  import tf_euler
  from euler_estimator.python.base_estimator import BaseEstimator
label_desc: Use image node_id 4 for calling the EdgeEstimator obj's underlying member method code with example usage: obj.transfer_embedding(source, emb) and returns: source,  emb, source,  emb, source,  emb