code: def __init__(self, module_cls, *args, **kwargs):
      super().__init__()
      self.re_module = module_cls(*args, **kwargs)
      self.im_module = module_cls(*args, **kwargs)
  
imports: import functools
  from functools import partial
  import numpy
  import torch
  from torch import Tensor, nn
  from torch.nn.modules.batchnorm import _BatchNorm
label_desc: Use image node_id 1 to create a new OnReIm object from inherited base classes: nn with example: obj = OnReIm(module_cls)