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