EinopsAst

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

EinopsAst function

<dl><dt>input_axes</dt>
<dd>List of axis nodes for the input pattern, or
<code>OneSidedAstNode()</code></dd>
<dt>output_axes</dt>
<dd>List of axis nodes for the input pattern, or
<code>OneSidedAstNode()</code></dd>
<dt>src</dt>
<dd>List with start position covering the full pattern</dd></dl>

Arguments

Create an EinopsAst root node — EinopsAst

<dl>

<dt>input_axes</dt>
<dd>List of axis nodes for the input pattern, or
<code>OneSidedAstNode()</code></dd>


<dt>output_axes</dt>
<dd>List of axis nodes for the input pattern, or
<code>OneSidedAstNode()</code></dd>


<dt>src</dt>
<dd>List with start position covering the full pattern</dd>

</dl>

EinopsAst: Create an EinopsAst root node

Description

Usage

Value

Arguments