code: def drop_feature(x, drop_prob):
      drop_mask = (
          th.empty(
              (x.size(1),), dtype=th.float32, device=x.device
          ).uniform_(0, 1)
          < drop_prob
      )
      x = x.clone()
      x[:, drop_mask] = 0
  
      return x
  
imports: import dgl
  import numpy
  import torch
label_desc: Use image node_id 2 for calling a global function with example usage: drop_feature(x, drop_prob) and returns: x