code: def forward(self, x):
      N, C, H, W = x.size()
      x = x.view(
          N, C, H // self.bs, self.bs, W // self.bs, self.bs
      )  # (N, C, H//bs, bs, W//bs, bs)
      x = x.permute(
          0, 3, 5, 1, 2, 4
      ).contiguous()  # (N, bs, bs, C, H//bs, W//bs)
      x = x.view(
          N, C * self.bs * self.bs, H // self.bs, W // self.bs
      )  # (N, C*bs^2, H//bs, W//bs)
      return x
  
file_path: pytorch-image-models/timm/layers/space_to_depth.py
imports: import torch
  import torch.nn
label_desc: Use image node_id 2 for calling the SpaceToDepth obj's underlying member method code with example usage: obj.forward(x) and returns: x