arg_max

Computes the indices of the max elements of the input tensor's element along the provided axis.

---

Parameters

• axis : [int] The axis to reduce.
• keepdims : [bool] Keep the reduced dimension or not.
• select_last_index : [bool] Whether to select the last or the first index if the max value apears in multiple indices. 0: First, 1: Last.