class_docstr: not again...
class_docstr_tok: ["not","again","..."]
docstr: For cases where shapes of Q and K do not match.
docstr_tok: ["For","cases","where","shapes","of","Q","and","K","do","not","match","."]
code: def single_forward(self, inputs: torch.Tensor):
      """For cases where shapes of Q and K do not match."""
      cos, sin = (
          self.cos_cached[: inputs.shape[0]],
          self.sin_cached[: inputs.shape[0]],
      )
      return inputs * cos + self.rotate_half(inputs) * sin
  
imports: import torch
  import math
  from typing import Tuple
  from einops import repeat
label_desc: Use image node_id 6 for calling the RotarySanityCheck obj's underlying member method code with example usage: obj.single_forward(inputs) and returns: unknown