File size: 422 Bytes
9c6594c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
# imports can use EinopsError class
# ruff: noqa: E402
__author__ = "Alex Rogozhnikov"
__version__ = "0.8.1"
class EinopsError(RuntimeError):
"""Runtime error thrown by einops"""
pass
__all__ = ["rearrange", "reduce", "repeat", "einsum", "pack", "unpack", "parse_shape", "asnumpy", "EinopsError"]
from .einops import rearrange, reduce, repeat, einsum, parse_shape, asnumpy
from .packing import pack, unpack
|