`BookProof.ChapterAttentionQKCircuit.qkMatrix_gauge` {A B : Matrix (Fin d) (Fin d) ℝ} (hAB : Aᵀ * B = 1) (WQ WK : Matrix (Fin d) (Fin n) ℝ) : qkMatrix (A * WQ) (B * WK) = qkMatrix
ProvedBookProof.ChapterAttentionQKCircuit.qkMatrix_gaugetheoremstimepiece
Prove the following Lean 4 theorem from ChapterAttentionQKCircuit.
BookProof.ChapterAttentionQKCircuit.qkMatrix_gauge {A B : Matrix (Fin d) (Fin d) ℝ} (hAB : Aᵀ * B = 1) (WQ WK : Matrix (Fin d) (Fin n) ℝ) : qkMatrix (A * WQ) (B * WK) = qkMatrix WQ WK
Formalization note: Lean 4 identifier BookProof.ChapterAttentionQKCircuit.qkMatrix_gauge.
Preamble
-- Generated from ChapterAttentionQKCircuit.lean — theorem BookProof.ChapterAttentionQKCircuit.qkMatrix_gauge
import Definitions.Def_ChapterSoftmaxSharpness
import Definitions.Def_ChapterSoftmaxOrder
import Mathlib
import Definitions.Def_ChapterAttentionQKCircuit
open BookProof.ChapterAttentionQKCircuit
open scoped BigOperators
noncomputable section
open Matrix BookProof.ChapterSoftmaxSharpness BookProof.ChapterSoftmaxOrder
variable {d n m : ℕ} {E : Type*} [NormedAddCommGroup E] [NormedSpace ℝ E]Formal statement
theorem BookProof.ChapterAttentionQKCircuit.qkMatrix_gauge {A B : Matrix (Fin d) (Fin d) ℝ} (hAB : Aᵀ * B = 1)
(WQ WK : Matrix (Fin d) (Fin n) ℝ) :
qkMatrix (A * WQ) (B * WK) = qkMatrix WQ WK := by sorry