espnet3.systems.f5tts.src.rotary.apply_rotary_pos_emb

source

espnet3.systems.f5tts.src.rotary.apply_rotary_pos_emb(t: Tensor, freqs: Tensor, scale=1) → Tensor

Apply rotary embeddings to t, forced to fp32.

• Parameters:
  • t – Queries or keys, [b, h, n, d] or [b, n, d].
  • freqs – Rotation angles, [b, n, rot_dim].
  • scale – xpos scale; always 1 on the path F5 uses.
• Returns:t rotated, cast back to its original dtype.