class_docstr: SpatialFc module that take features with shape of (B,C,*) as input.
  Args:
      spatial_shape (tuple[int]): spatial shape
code: def forward(self, x):
      # input shape like [B, C, H, W]
      shape = x.shape
      x = torch.flatten(x, start_dim=2)  # [B, C, H*W]
      x = self.fc(x)  # [B, C, H*W]
      x = x.reshape(*shape)  # [B, C, H, W]
      return x
  
imports: from torch import nn
  import torch
  from typing import Sequence
  from functools import partial, reduce
label_desc: Use image node_id 2 for calling the SpatialFc obj's underlying member method code with example usage: obj.forward(x) and returns: x