SpectralOperation

class cosmax.spectral_ops.SpectralOperation(elements: int, size: float = 1.0)

Bases: object

Base class for spectral operations

Parameters:
  • elements – number of grid points in each dimension

  • size – size of the box in real space

elements

number of grid points in each dimension

Type:

int

size

size of the box in real space

Type:

float

frequencies

frequencies in fourier space

Type:

jax.Array

real_frequencies

real frequencies in fourier space

k_mag

magnitude of the wavenumber