enerzyme.models.physnet.interaction.DenseLayer

enerzyme.models.physnet.interaction.DenseLayer(dim_feature_in: int, dim_feature_out: int, 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 | str | None = None, initial_bias: Tensor | ndarray | str | None = None, use_bias: bool = True) → DenseLayer