enerzyme.models.layers.gradient.ForceLayer#

class enerzyme.models.layers.gradient.ForceLayer[source]#

Bases: BaseFFLayer

__init__() → None[source]#

get_Fa(E: Tensor, Ra: Tensor) → Tensor[source]#