class_docstr: not again...
class_docstr_tok: ["not","again","..."]
code: def forward(self, query_layer: torch.Tensor, key_layer: torch.Tensor):
      # cos, sin = self.get_cos_sin_cache(key_layer)
      # cos, sin = (cos[offset : query_layer.shape[0] + offset, ...], sin[offset : query_layer.shape[0] + offset, ...])
      cos, sin = self.cos_cached, self.sin_cached
      return (query_layer * cos) + (
          self.rotate_half(query_layer) * sin
      ), (key_layer * cos) + (self.rotate_half(key_layer) * sin)
  
imports: import torch
  import math
  from typing import Tuple
  from einops import repeat
label_desc: Use image node_id 4 for calling the RotarySanityCheck obj's underlying member method code with example usage: obj.forward(query_layer, key_layer) and returns: unknown,  unknown