Kernels:

kernels-community
/

einops

Trusted publisher

Kernel card Files Files and versions

einops / build /torch-cpu /layers /keras.py

kernels-bot's picture

Uploaded using `kernel-builder`.

cb47c7f verified about 19 hours ago

212 Bytes

	__author__ = "Alex Rogozhnikov"

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

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