code: def _dropout_unmapping_fn(
      module: torch.nn.Module,
  ) -> Optional[nn.Module]:
      new_module: Optional[nn.Module] = None
      if isinstance(module, Dropout):
          new_module = torch.nn.Dropout(
              p=module.p, inplace=module.inplace
          )
      elif isinstance(module, Dropout2d):
          new_module = torch.nn.Dropout2d(
              p=module.p, inplace=module.inplace
          )
      return new_module
  
file_path: baal/baal/bayesian/dropout.py
imports: from typing import Optional
  import torch
  from torch import nn
  from torch.nn import functional
  from torch.nn.modules.dropout import _DropoutNd
  from baal.bayesian.common import BayesianModule, _patching_wrapper
label_desc: Use image node_id 4 for calling a global function with example usage: _dropout_unmapping_fn(module) and returns: new_module