Skip to content


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


  • 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.