Package index

Core Einops Operations

reduce() einops.reduce()

Rearrangement and reduction in one step

rearrange() einops.rearrange()

Reader-friendly smart element reordering for multidimensional tensors.

einops.repeat() `repeat`()

Allows reordering elements and repeating them in arbitrary combinations.

einop()

Perform Einstein-style tensor operations

Utilities

parse_shape()

Parse a tensor shape to dictionary mapping axes names to their lengths.

Image Tensor class

as_image_tensor() image_tensor()

Image Tensor: A thin wrapper around 2-4D arrays

einops_image

Example 4D Image Tensor for Einops