Skip to contents

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() as.cimg(<image_tensor>) `[`(<image_tensor>) plot(<image_tensor>) print(<image_tensor>)
Image Tensor: A thin wrapper around 2-4D arrays
einops_image
Example 4D Image Tensor for Einops