Man pages for einops
Flexible, Powerful, and Readable Tensor Operations

AddOnlyOrderedMapCreate an 'AddOnlyOrderedMap' instance
appendVector Merging
apply_recipeApply a TransformRecipe to a Tensor
as_axis_namesConvert an AstNode into an 'AxisNames()' object
as_iterablesConvert an object to a list of iterables (lists)
AxisNamesConstruct an instance of an 'AxisNames' class
BackendRegistrySingleton Backend Registry, managing all available backends.
ConstantAstNodeCreate a ConstantAstNode
create_execution_planReconstruct all actual parameters using shape.
einopPerform Einstein-style tensor operations
EinopsAstCreate an EinopsAst root node
EinopsBackendBase Backend Class for Einops Tensor Operations
EinopsExecutionPlanConstructor for an Execution Plan ('CookedRecipe' in the...
einops_imageExample 4D Image Tensor for Einops
einops-packageeinops: Flexible, Powerful, and Readable Tensor Operations
einops.repeatAllows reordering elements and repeating them in arbitrary...
EinopsTokenSequenceEinopsTokenSequence constructor
EllipsisAstNodeCreate an EllipsisAstNode
expand_ellipsisExpand ellipses of an EinopsAst
find_top_level_arrow_indexFind the arrow in a token sequence
get_backendMain function to detect and return backend
get_identifiersGiven a 'OneSidedAstNode' object, get unique identifiers
get_ordered_axis_namesGet the ordered axis names from a OneSidedAstNode, removing...
get_ungrouped_nodesget flat list of all AstNodes in order, including children of...
GroupAstNodeCreate a GroupAstNode
image_tensorImage Tensor: A thin wrapper around 2-4D arrays
lexLexically analyze einops pattern into structured tokens
NamedAxisAstNodeCreate a NamedAxisAstNode
OneSidedAstNodeCreate a OneSidedAstNode (wrapper for input/output axes...
parse_axes_iterParse a sequence of axis tokens
parse_einops_astParse einops pattern into AST
parse_onesided_astParse a one-sided einops pattern into OneSidedAstNode
parse_shapeParse a tensor shape to dictionary mapping axes names to...
pipePipe operator
pprintPretty Print
prepare_transformation_recipeCreate the Transformation Recipe for an einops call
rearrangeReader-friendly smart element reordering for multidimensional...
reduceRearrangement and reduction in one step
register_backendRegister a new backend for a tensor type
reprPython-like Representation of Objects as Strings
tail.GroupAstNodeGet the last n children of a GroupAstNode as a list of...
thunkSimple thunk: wraps an input in a no-argument function
TransformRecipeTransformRecipe S3 constructor
UnderscoreAstNodeCreate an UnderscoreAstNode
validate_reduction_operationSyntactically Validate the reduction operation in the einops...
einops documentation built on Sept. 9, 2025, 5:29 p.m.