enerzyme.models.layers.scalar_embedding.ScalarEmbedding

class enerzyme.models.layers.scalar_embedding.ScalarEmbedding(embed_field: str)

Bases: BaseFFLayer

__init__(embed_field: str) → None

get_output(**relevant_input: Dict[str, Tensor]) → Dict[str, Tensor]