10 lines
217 B
Python
10 lines
217 B
Python
__author__ = "Alex Rogozhnikov"
|
|
|
|
from einops.layers.tensorflow import EinMix, Rearrange, Reduce
|
|
|
|
keras_custom_objects = {
|
|
Rearrange.__name__: Rearrange,
|
|
Reduce.__name__: Reduce,
|
|
EinMix.__name__: EinMix,
|
|
}
|