BatchInputSizes

class cubie.outputhandling.output_sizes.BatchInputSizes(initial_values: Tuple[int, int] = (1, 1), parameters: Tuple[int, int] = (1, 1), driver_coefficients: Tuple[int | None, int, int | None] = (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

default (1, 1, 1) Shape of the driver coefficient array as (num_segments, num_drivers, polynomial_degree).

Type:

tuple[int or None, int, int or None],

driver_coefficients: Tuple[int | None, int, int | None]

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]