Extends the Argmax operator to handle dynamic input shapes. Only shape function changes
Attach a file by drag & drop or click to upload
