__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

__author__ = 'Alex Rogozhnikov'
__version__ = '0.4.1'


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


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

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

Include einsum in main library
Include einsum in main library

  __author__ = 'Alex Rogozhnikov'
  __version__ = '0.4.1'
  
  
  class EinopsError(RuntimeError):
      """ Runtime error thrown by einops """
      pass
  
  
- __all__ = ['rearrange', 'reduce', 'repeat', 'parse_shape', 'asnumpy', 'EinopsError']
+ __all__ = ['rearrange', 'reduce', 'repeat', 'einsum',
+            'parse_shape', 'asnumpy', 'EinopsError']
  
- from .einops import rearrange, reduce, repeat, parse_shape, asnumpy
+ from .einops import rearrange, reduce, repeat, einsum, parse_shape, asnumpy
  
Include einsum in main library
## Code Before:
__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

## Instruction:
Include einsum in main library
## Code After:
__author__ = 'Alex Rogozhnikov'
__version__ = '0.4.1'


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


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

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