axis_batch package

class axis_batch.AxisBatch(a: TArray, /, *, axis: int | Sequence[int], size: int)

Bases: Generic[TArray]

Batch the array along the given axis.

send(result: TArray, /) → None

Add the batched array to the results.

Parameters:

result (TArray) – The batched array.

property value: TArray

Return the concentrated batched arrays.

Returns:

The concentrated batched arrays.

Return type:

TArray