get_ungrouped_nodes

get flat list of all AstNodes in order, including children of
GroupAstNodes.

internal

Perform tensor operations using a concise yet expressive syntax inspired by the Python library of the same name.
Reshape, rearrange, and combine multidimensional arrays for scientific computing, machine learning, and data analysis.
Einops simplifies complex manipulations, making code more maintainable and intuitive.
The original implementation is demonstrated in Rogozhnikov (2022) <https://openreview.net/forum?id=oapKSVM2bcj>.

Qile Yang

einops

Flexible, Powerful, and Readable Tensor Operations

get_ungrouped_nodes function

<dl><dt>ast</dt>
<dd>An AstNode object</dd></dl>

Arguments

get flat list of all AstNodes in order, including children of
GroupAstNodes. — get_ungrouped_nodes

<dl>

<dt>ast</dt>
<dd>An AstNode object</dd>

</dl>

get_ungrouped_nodes: get flat list of all AstNodes in order, including children of GroupAstNodes.

Description

Usage

Value

Arguments