/ open-webui.git / app / env / 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, }