Skip to content

Energy_function

Contents

Classes

  • AckleyEnergy - Energy function for the Ackley function.
  • DoubleWellEnergy - Energy function for a double well potential. E(x) = h * Σ((x²-1)²) where h is the barrier height.
  • EnergyFunction - Abstract base class for energy functions (Potential Energy E(x)).
  • GaussianEnergy - Energy function for a Gaussian distribution. E(x) = 0.5 * (x-μ)ᵀ Σ⁻¹ (x-μ).
  • HarmonicEnergy - Energy function for a harmonic oscillator. E(x) = 0.5 * k * Σ(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.energy_function