__init__.py 438 B

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