einhighlightHighlights axes in einops patterns so matching axes share the same color, and unmatched axes (added or removed) are highlighted in red. |
einhighlightHighlights axes in einops patterns so matching axes share the same color, and unmatched axes (added or removed) are highlighted in red. |