import tensorflow as tf

    neighbor, weight, _ = get_full_neighbor(nodes, hop_edge_types)
    next_nodes, next_idx = tf.unique(neighbor.values, out_idx=tf.int64)
    next_indices = tf.stack([neighbor.indices[:, 0], next_idx], 1)
    next_values = weight.values
