BatchInputSizes

class cubie.outputhandling.output_sizes.BatchInputSizes(initial_values: Tuple[int, int] = (1, 1), parameters: Tuple[int, int] = (1, 1), driver_coefficients: Tuple[int, int, int] = (1, 1, 1))

Bases: ArraySizingClass

Input array sizes for batch integration runs.

This class specifies the sizes of input arrays needed for batch processing, including initial conditions, parameters, and forcing terms.

initial_values

Shape of initial values array as (n_states, n_runs).

Type:

tuple[int, int], default (1, 1)

parameters

Shape of parameters array as (n_parameters, n_runs).

Type:

tuple[int, int], default (1, 1)

driver_coefficients

Shape of the driver coefficient array as (num_segments, num_drivers, order + 1), pinned from the driver interpolator’s compiled layout.

Type:

tuple[int, int, int], default (1, 1, 1)

driver_coefficients: Tuple[int, int, int]

classmethod from_solver(solver_instance: BatchSolverKernel) → BatchInputSizes

Create batch input shapes based on solver metadata.

Parameters:

solver_instance – Batch solver kernel exposing num_runs and system sizes.

Returns:

Input array dimensions for the batch run.

Return type:

BatchInputSizes

initial_values: Tuple[int, int]

parameters: Tuple[int, int]