Skip to content
heliaAOT
HELIA HUB

AirSplitOptions

Machine-readable model

class

AirSplitOptions

Python

Options for the SPLIT operator.

helia_aot/air/options.py:653

AirSplitOptions()

Options for the SPLIT operator.

attribute

axis

Python

helia_aot/air/options.py:663

axis: int = Field(0, description='Axis along which to split the input tensor')

Axis along which to split the input tensor (default 0).

attribute

split_lengths

Python

helia_aot/air/options.py:664

split_lengths: list[int] = Field(default_factory=list, description='List of lengths for each split')

List of lengths for each split (default empty list).