`frame_axis`

python

frame_axis(observation_shape_order: Literal['hwf', 'fhw']) -> int

Return the frame axis index for the configured observation layout.

Parameters:

observation_shape_order (Literal['hwf', 'fhw']) – The configured observation layout.

Returns:

int – The axis holding the frame-stack dimension.