code: def _select_pinsage_neighbors(src, dst, num_samples_per_node, k):
      """Determine the neighbors for PinSAGE algorithm from the given random walk traces.
  
      This is fusing ``to_simple()``, ``select_topk()``, and counting the number of occurrences
      together.
      """
      src = F.to_dgl_nd(src)
      dst = F.to_dgl_nd(dst)
      src, dst, counts = _CAPI_DGLSamplingSelectPinSageNeighbors(
          src, dst, num_samples_per_node, k
      )
      src = F.from_dgl_nd(src)
      dst = F.from_dgl_nd(dst)
      counts = F.from_dgl_nd(counts)
      return (src, dst, counts)
  
file_path: dgl/python/dgl/sampling/pinsage.py
imports: import numpy
  from ..None import backend, convert, utils
  from .._ffi.function import _init_api
  from .randomwalks import random_walk
label_desc: Use image node_id 1 for calling a global function with example usage: _select_pinsage_neighbors(src, dst, num_samples_per_node, k) and returns: src,  dst,  counts