enerzyme.models.layers.atom_embedding.BaseAtomEmbedding#

class enerzyme.models.layers.atom_embedding.BaseAtomEmbedding(max_Za, dim_embedding)[source]#

Bases: BaseFFLayer

__init__(max_Za, dim_embedding) → None[source]#

get_atom_embedding(Za: Tensor) → Tensor[source]#

abstractmethod get_embedding(Za: Tensor) → Tensor[source]#