apply_recipe

This function applies a TransformRecipe to a tensor, performing rearrangement
and reduction as specified by the recipe.
TODO docs for which parts of the compilation pipeline this is.

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

apply_recipe function

<dl><dt>backend</dt>
<dd>The <code>EinopsBackend()</code> to use for tensor operations.</dd>
<dt>recipe</dt>
<dd>A <code>TransformRecipe()</code> that specifies how to transform the
tensor.</dd>
<dt>tensor</dt>
<dd>A tensor to be transformed</dd>
<dt>reduction_type</dt>
<dd>A character of length 1 that specifies the type of
reduction to apply.</dd>
<dt>axes_lengths</dt>
<dd>TODO check - I think its just an integer vector?</dd></dl>

Arguments

Apply a TransformRecipe to a Tensor — apply_recipe

<dl>

<dt>backend</dt>
<dd>The <code>EinopsBackend()</code> to use for tensor operations.</dd>


<dt>recipe</dt>
<dd>A <code>TransformRecipe()</code> that specifies how to transform the
tensor.</dd>


<dt>tensor</dt>
<dd>A tensor to be transformed</dd>


<dt>reduction_type</dt>
<dd>A character of length 1 that specifies the type of
reduction to apply.</dd>


<dt>axes_lengths</dt>
<dd>TODO check - I think its just an integer vector?</dd>

</dl>

apply_recipe: Apply a TransformRecipe to a Tensor

Description

Usage

Value

Arguments