Split

class baikal.steps.Split(indices_or_sections, axis=-1, name=None)

Bases: baikal._core.step.Step

Step for splitting arrays.

Parameters
  • indices_or_sections – If an integer (N) is passed, the array will be divided into N equal arrays along axis. If an 1-D array of sorted integers is passed, the entries indicate where along axis the array is split.

  • axis – The axis on where to split the array (default is -1, the last axis).

  • name – Name of the step (optional). If no name is passed, a name will be automatically generated.