Make FactoredMatrix compatible with tensor-like arguments #599
Draft
JasonGross wants to merge 2 commits intoTransformerLensOrg:devfrom JasonGross:patch-1
+6-6
FactoredMatrix compatible with tensor-like arguments #599
