__init__.py 422 B

1234567891011121314151617
  1. # imports can use EinopsError class
  2. # ruff: noqa: E402
  3. __author__ = "Alex Rogozhnikov"
  4. __version__ = "0.8.1"
  5. class EinopsError(RuntimeError):
  6. """Runtime error thrown by einops"""
  7. pass
  8. __all__ = ["rearrange", "reduce", "repeat", "einsum", "pack", "unpack", "parse_shape", "asnumpy", "EinopsError"]
  9. from .einops import rearrange, reduce, repeat, einsum, parse_shape, asnumpy
  10. from .packing import pack, unpack