enerzyme.models.layers.electrostatics.AtomicCharge2DipoleLayer

class enerzyme.models.layers.electrostatics.AtomicCharge2DipoleLayer

Bases: BaseFFLayer

__init__() → None

get_M2(Qa: Tensor, Ra: Tensor, batch_seg: Tensor | None = None) → Tensor