keras_mml.layers.coreΒΆ

Core layers implemented by Keras-MML.

Classes

DenseMML

Dense layer without matrix multiplications.

PatchEmbedding

Turns image patches into vectors of fixed size.

TokenEmbedding

Turns positive integers (indices) into vectors of fixed size.