deeperwin.model.PauliNetEmbedding
- class deeperwin.model.PauliNetEmbedding(*args, **kwargs)[source]
Bases:
haiku._src.module.Module
- __init__(config: deeperwin.configuration.EmbeddingConfigDeepErwin1, mlp_config: deeperwin.configuration.MLPConfig, n_up, name=None)[source]
Initializes the current module with the given name.
Subclasses should call this constructor before creating other modules or variables such that those modules are named correctly.
- Parameters
name – An optional string name for the class. Must be a valid Python identifier. If
name
is not provided then the class name for the current instance is converted tolower_snake_case
and used instead.
Methods
__init__
(config, mlp_config, n_up[, name])Initializes the current module with the given name.