code: def mask_edge(graph, mask_prob):
      E = graph.num_edges()
  
      mask_rates = th.FloatTensor(np.ones(E) * mask_prob)
      masks = th.bernoulli(1 - mask_rates)
      mask_idx = masks.nonzero().squeeze(1)
      return mask_idx
  
imports: import dgl
  import numpy
  import torch
label_desc: Use image node_id 3 for calling a global function with example usage: mask_edge(graph, mask_prob) and returns: mask_idx