enerzyme.models.layers.reduce.ShallowEnsembleReduceLayer#

class enerzyme.models.layers.reduce.ShallowEnsembleReduceLayer(reduce_mean: List[str] = [], var: List[str] = [], std: List[str] = [], relative_energy: bool = False, train_only: bool = False, eval_only: bool = False, test_only: bool = False, test_exclude: bool = False)[source]#

Bases: BaseFFLayer

__init__(reduce_mean: List[str] = [], var: List[str] = [], std: List[str] = [], relative_energy: bool = False, train_only: bool = False, eval_only: bool = False, test_only: bool = False, test_exclude: bool = False) → None[source]#
get_output(**relevant_input: Dict[str, Tensor]) → Dict[str, Tensor][source]#