code: def find_indices_in(a, b):
      b_sorted, indices = torch.sort(b)
      sorted_indices = torch.searchsorted(b_sorted, a)
      sorted_indices[sorted_indices >= indices.shape[0]] = 0
      return indices[sorted_indices]
  
file_path: dgl/examples/pytorch/labor/ladies_sampler.py
imports: import dgl
  import dgl.function
  import torch
label_desc: Use image node_id 1 for calling a global function with example usage: find_indices_in(a, b) and returns: indices