code: def forward(self, sm_tokens, lg_tokens):
      for sm_enc, lg_enc, cross_attend in self.layers:
          sm_tokens, lg_tokens = sm_enc(sm_tokens), lg_enc(lg_tokens)
          sm_tokens, lg_tokens = cross_attend(sm_tokens, lg_tokens)
  
      return sm_tokens, lg_tokens
  
imports: import torch
  from torch import nn, einsum
  import torch.nn.functional
  from einops import rearrange, repeat
  from einops.layers.torch import Rearrange
label_desc: Use image node_id 2 for calling the MultiScaleEncoder obj's underlying member method code with example usage: obj.forward(sm_tokens, lg_tokens) and returns: sm_tokens,  lg_tokens