expand_ellipsis

Helper for <code>prepare_transformation_recipe()</code>.
This function expands each relevant ellipsis ast node in-place into a
sequence of <code>NamedAxisAstNode</code> nodes, where each node will have
a name like <code>"...1"</code>, <code>"...2"</code>, etc. and an empty <code>src</code> list.
Also does some further validation of the ellipses syntax using <code>ndim</code>.

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

expand_ellipsis function

<dl><dt>einops_ast</dt>
<dd>an EinopsAst</dd>
<dt>ndim</dt>
<dd>integer. Number of dimensions in the input tensor</dd></dl>

Arguments

Expand ellipses of an EinopsAst — expand_ellipsis

<dl>

<dt>einops_ast</dt>
<dd>an EinopsAst</dd>


<dt>ndim</dt>
<dd>integer. Number of dimensions in the input tensor</dd>

</dl>

expand_ellipsis: Expand ellipses of an EinopsAst

Description

Usage

Value

Arguments