enerzyme.models.physnet.interaction.OutputBlock

enerzyme.models.physnet.interaction.OutputBlock(dim_embedding: int, num_residual: int, activation_fn: Literal['shifted_softplus', 'swish'] = None, activation_params: Dict[Literal['dim_feature', 'initial_alpha', 'initial_beta', 'learnable'], int | float | bool] = {}, dropout_rate: float = 0.0, shallow_ensemble_size: int = 1) → ResidualMLP