Python wrappers
Public Python APIs implemented by this engine symbol.Index of the maximum value along an axis.
Parameters
aTensorImplPtrInput tensor.
axisintAxis to reduce.
keepdimsboolIf true, retain the reduced axis as size 1 in the output shape; otherwise remove it.
Returns
TensorImplPtrInteger tensor of dtype I64 containing the argmax indices.
Notes
Ties are broken by returning the lowest index. Non-differentiable.