AirArgMaxOptions

Machine-readable model

• AirArgMaxOptionsclassOptions for the ARGMAX operator.

class

AirArgMaxOptions

Python

Options for the ARGMAX operator.

helia_aot/air/options.py:563

AirArgMaxOptions()

Options for the ARG_MAX operator.

Float routes use a logical input axis (negative axes are supported). The legacy integer parser supplies its existing padded kernel axis.

attribute

axis

Python

helia_aot/air/options.py:571

axis: int = Field(0, description='Axis along which to compute the argmax')