Need argmax/argmin operator

[SunXuan90]

Is this a new feature, an improvement, or a change to existing functionality?

New Feature

How would you describe the priority of this feature request

Must have (e.g. DALI adoption is impossible due to lack in functionality).

Please provide a clear description of problem this feature solves

Object detection postprocessing, for example, slicing model output according to confidence, class, etc.

Feature Description

We already have dali.fn.reductions.max, just add argmax and argmin that returns index rather than value.

Describe your ideal solution

Add argmax.

Describe any alternatives you have considered

No response

Additional context

No response

Check for duplicates

• I have searched the open bugs/issues and have found no duplicates for this bug report

[JanuszL]

Hi @SunXuan90,

Thank you for bringing this to our attention. We will add this to our ToDo list.