code: def __init__(self, no_jit=False):
      super().__init__()
      if not no_jit:
          self.op = SpaceToDepthJit()
      else:
          self.op = SpaceToDepth()
  
file_path: pytorch-image-models/timm/layers/space_to_depth.py
imports: import torch
  import torch.nn
label_desc: Use image node_id 1 to create a new SpaceToDepthModule object from inherited base classes: nn with example: obj = SpaceToDepthModule(no_jit)