code: def sample_node_with_src(src_nodes, count):
      """
      for each src node, sample "n" nodes with the same type
  
      Args:
        src_nodes: A 1-d `Tensor` of `int64`
        n: A scalar value of int
      Returns:
        A 2-dim Tensor, the first dim should be equal to dim of src_node.
        The second dim should be equal to n.
      """
      types = base._LIB_OP.get_node_type(src_nodes)
      return base._LIB_OP.sample_n_with_types(count, types)
  
file_path: euler/tf_euler/python/euler_ops/sample_ops.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import ctypes
  import os
  import tensorflow
  from tf_euler.python.euler_ops import base
  from tf_euler.python.euler_ops import type_ops
label_desc: Use image node_id 4 for calling a global function with example usage: sample_node_with_src(src_nodes, count) and returns: base