parse_einops_ast: Parse einops pattern into AST

In einops: Flexible, Powerful, and Readable Tensor Operations

Parse einops pattern into AST

Description

In the python implementation, they preprocess all 1's into () and then parse it into the equivalent of a GroupAstNode. Here, we won't do that

Usage

parse_einops_ast(tokens)

Arguments

tokens

EinopsTokenSequence object from the lexer

Value

EinopsAst object

einops documentation built on Sept. 9, 2025, 5:29 p.m.