code: def __init__(self, block_size=4):
      super().__init__()
      assert block_size == 4
      self.bs = block_size
  
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 SpaceToDepth object from inherited base classes: nn with example: obj = SpaceToDepth(block_size)