class_docstr: Rotary position embedding
  
  NOTE: This is my initial attempt at impl rotary embedding for spatial use, it has not
  been well tested, and will likely change. It will be moved to its own file.
  
  The following impl/resources were referenced for this impl:
  * https://github.com/lucidrains/vit-pytorch/blob/6f3a5fcf0bca1c5ec33a35ef48d97213709df4ba/vit_pytorch/rvt.py
  * https://blog.eleuther.ai/rotary-embeddings/
code: def get_embed(self, shape: Optional[List[int]] = None):
      if self.bands is not None:
          # rebuild embeddings every call, use if target shape changes
          assert shape is not None
          return build_rotary_pos_embed(
              shape,
              self.bands,
              in_pixels=self.in_pixels,
          )
      else:
          return self.pos_embed_sin, self.pos_embed_cos
  
imports: import math
  from typing import List, Tuple, Optional, Union
  import torch
  from torch import nn
  from .trace_utils import _assert
label_desc: Use image node_id 2 for calling the RotaryEmbedding obj's underlying member method code with example usage: obj.get_embed(shape) and returns: build_rotary_pos_embed, self,  self