__author__ = 'Alex Rogozhnikov'
__version__ = '0.4.1'


class EinopsError(RuntimeError):
    """ Runtime error thrown by einops """
    pass


__all__ = ['rearrange', 'reduce', 'repeat', 'parse_shape', 'asnumpy', 'EinopsError']

from .einops import rearrange, reduce, repeat, parse_shape, asnumpy

@@ -179,16 +179,37 @@
 repeat',
+ 'einsum',%0A          
  'parse_
@@ -289,16 +289,24 @@
  repeat,
+ einsum,
  parse_s

Include einsum in main library