code: def img_siren_head(
      size: int, out_channels: int
  ) -> Callable[[Tensor], Tensor]:
      def _head(t: Tensor) -> Tensor:
          t = t.view(-1, size, size, out_channels)
          t = t.permute(0, 3, 1, 2)
          return t
  
      return _head
  
imports: import math
  import torch
  import torch.nn
  import torch.nn.functional
  from torch import Tensor
  from typing import Tuple
  from typing import Union
  from typing import Callable
  from typing import Optional
  from cftool.types import tensor_dict_type
  from ...constants import LABEL_KEY
  from ...misc.toolkit import insert_intermediate_dims
  from ...modules.blocks import Lambda
  from ...modules.blocks import Activation
  from ...modules.blocks import ChannelPadding
label_desc: Use image node_id 2 for calling a global function with example usage: img_siren_head(size, out_channels) and returns: _head, t