code: def _split_heads(t: torch.Tensor, B: int, S: int, H: int, Hs: int):
      return t.view(B, S, H, Hs).transpose(1, 2)
  
file_path: xformers/xformers/components/multi_head_dispatch.py
imports: import logging
  from dataclasses import asdict, dataclass
  from typing import Optional, Tuple
  import torch
  import torch.nn
  from torch.nn.init import constant_
  from xformers.components.attention import Attention
  from xformers.components.input_projection import InputProjection, InputProjectionConfig
  from xformers.components.positional_embedding import RotaryEmbedding
label_desc: Use image node_id 2 for calling a global function with example usage: _split_heads(t, B, S, H, Hs) and returns: t