code: def backward(ctx, out_gradients: torch.Tensor):
      out_gradients = out_gradients.contiguous()
      global random_ltd_module
      if random_ltd_module is None:
          random_ltd_module = RandomLTDBuilder().load()
      (sorted_indices,) = ctx.saved_tensors
      batch_first = ctx.batch_first
  
      ret_val = random_ltd_module.token_gather(
          out_gradients, sorted_indices, batch_first
      )
      return out_gradients, ret_val, None, None
  
file_path: deepspeed/deepspeed/ops/random_ltd/dropping_utils.py
imports: import torch
  from deepspeed.ops.op_builder import RandomLTDBuilder
label_desc: Use image node_id 2 for calling the ScatterTokens obj's underlying member method code with example usage: obj.backward(ctx, out_gradients) and returns: out_gradients,  ret_val,  None,  None