Skip to content

Torchebm > Core > Base_energy_function

Contents

Classes

  • AckleyEnergy - Energy function for the Ackley function.
  • BaseEnergyFunction - Abstract base class for energy functions (Potential Energy \(E(x)\)).
  • DoubleWellEnergy - Energy function for a double well potential. \( E(x) = h Σ(x²-1)² \) where h is the barrier height.
  • GaussianEnergy - Energy function for a Gaussian distribution. \(E(x) = 0.5 (x-μ)ᵀ Σ⁻¹ (x-μ)\).
  • HarmonicEnergy - Energy function for a harmonic oscillator. \(E(x) = 0.5 \cdot n\_steps \cdot Σ(x²)\).
  • RastriginEnergy - Energy function for the Rastrigin function.
  • RosenbrockEnergy - Energy function for the Rosenbrock function. \(E(x) = (a-x₁)² + b·(x₂-x₁²)²\).

API Reference

torchebm.core.base_energy_function