code: def apply_rot_embed(x: torch.Tensor, sin_emb, cos_emb):
      return x * cos_emb + rot(x) * sin_emb
  
imports: import math
  import warnings
  from collections import OrderedDict
  from typing import List, Optional, Tuple, Union
  import torch
  import torch.nn
  from torch import nn
  from lavis.models.clip_models.utils import freeze_batch_norm_2d
label_desc: Use image node_id 6 for calling a global function with example usage: apply_rot_embed(x, sin_emb, cos_emb) and returns: unknown