The input
The axis along which to perform the reduction. Negative values means indexing from right to left. Requires axis to be set as int, because global reduction is not supported yet.
If this is set to True
, the reduced axis is left in the result as dimension with size one.
This Param Object is specifically used for argmax