code: def cct_2(*args, **kwargs):
      return _cct(
          num_layers=2,
          num_heads=2,
          mlp_ratio=1,
          embedding_dim=128,
          *args,
          **kwargs,
      )
  
imports: import torch
  from torch import nn, einsum
  import torch.nn.functional
  from einops import rearrange, repeat
label_desc: Use image node_id 4 for calling a global function with example usage: cct_2() and returns: _cct