18 lines
438 B
Python
18 lines
438 B
Python
# imports can use EinopsError class
|
|
# ruff: noqa: E402
|
|
|
|
__author__ = "Alex Rogozhnikov"
|
|
__version__ = "0.8.2"
|
|
|
|
|
|
class EinopsError(RuntimeError):
|
|
"""Runtime error thrown by einops"""
|
|
|
|
pass # noqa: PIE790
|
|
|
|
|
|
__all__ = ["EinopsError", "asnumpy", "einsum", "pack", "parse_shape", "rearrange", "reduce", "repeat", "unpack"]
|
|
|
|
from .einops import asnumpy, einsum, parse_shape, rearrange, reduce, repeat
|
|
from .packing import pack, unpack
|