venv/lib/python3.11/site-packages/einops/layers/keras.py

__author__ = "Alex Rogozhnikov"

from ..layers.tensorflow import Rearrange, Reduce, EinMix

keras_custom_objects = {
    Rearrange.__name__: Rearrange,
    Reduce.__name__: Reduce,
    EinMix.__name__: EinMix,
}