Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
repeat
(includes repeat/tiling logic, copying along a new dimension)grayscale = reduce(image, 'h w 3 -> h w', 'mean')
image_with_identical_channels = repeat(grayscale, 'h w -> h w 3')
keras
layers are deprecated in favor of tf.keras
layersThanks to contributors
This release introduces einops, as well as its notion.
Initial release API: Operations (ops)
einops.rearrange
and einops.reduce
Auxiliary
einops.asnumpy
and einops.parse_shape
Layers (for chainer, gluon, keras and torch)
Rearrange
and Reduce
Supported frameworks: