code: def state_dict(module, destination=None, prefix="", keep_vars=False):
      destination = torch.nn.Module.state_dict(
          module,
          destination=destination,
          prefix=prefix,
          keep_vars=keep_vars,
      )
  
      for key in destination.keys():
          if "para_16" in key:
              destination.pop(key)
  
      return destination
  
file_path: lightseq/lightseq/csrc/pytorch/util.py
imports: import math
  from typing import Callable
  import numpy
  import torch
  import torch.nn.functional
label_desc: Use image node_id 5 for calling a global function with example usage: state_dict(module, destination, prefix, keep_vars) and returns: destination