import tensorflow as tf

  type_list = []
  for hop_edge_types, count in zip(edge_types, counts):
    neighbors, weights, types = sample_neighbor(
        neighbors_list[-1], hop_edge_types, count, default_node=default_node)
    neighbors_list.append(tf.reshape(neighbors, [-1]))
    weights_list.append(tf.reshape(weights, [-1]))
    type_list.append(tf.reshape(weights, [-1]))
  return neighbors_list, weights_list, type_list
