enerzyme.models.layers.scalar_embedding.ScalarDenseEmbedding

class enerzyme.models.layers.scalar_embedding.ScalarDenseEmbedding(dim_embedding: int, embed_field: str, activation_fn: Literal['shifted_softplus', 'swish'] | None = None, activation_params: Dict[Literal['dim_feature', 'initial_alpha', 'initial_beta', 'learnable'], int | float | bool] = {}, initial_weight: Tensor | ndarray | Literal['semi_orthogonal_glorot', 'orthogonal', 'zero', 'xavier_uniform'] = 'orthogonal', initial_bias: Tensor | ndarray | Literal['zero'] = 'zero', use_bias: bool = True)

Bases: ScalarEmbedding

__init__(dim_embedding: int, embed_field: str, activation_fn: Literal['shifted_softplus', 'swish'] | None = None, activation_params: Dict[Literal['dim_feature', 'initial_alpha', 'initial_beta', 'learnable'], int | float | bool] = {}, initial_weight: Tensor | ndarray | Literal['semi_orthogonal_glorot', 'orthogonal', 'zero', 'xavier_uniform'] = 'orthogonal', initial_bias: Tensor | ndarray | Literal['zero'] = 'zero', use_bias: bool = True) → None