desc.integrals.Bounce2D.reshape

static Bounce2D.reshape(grid, f)Source

Reshape arrays for acceptable input to integrate.

Parameters:

• grid (Grid) – Tensor-product grid in (ρ, θ, ζ).

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

Returns:

f (jnp.ndarray) – Shape (num rho, num zeta, num theta). Reshaped data which may be given to integrate.