desc.integrals.Bounce1D.reshape

static Bounce1D.reshape(grid, f)Source

Reshape arrays for acceptable input to integrate.

Parameters:

• grid (Grid) – Tensor-product grid in (ρ, α, ζ) Clebsch coordinates. The ζ coordinates (the unique values prior to taking the tensor-product) must be strictly increasing.

• f (jnp.ndarray) – Data evaluated on grid.

Returns:

f (jnp.ndarray) – Shape (num ρ, num α, num ζ). Reshaped data which may be given to integrate.