enerzyme.models.layers.atom_embedding.RandomAtomEmbedding#

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

Bases: BaseAtomEmbedding

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

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

property weight: Tensor#