function argMax_(x, axis = 0) {
    const $x = (0, _tensorUtilEnv.convertToTensor)(x, "x", "argMax");
    const inputs = {
        x: $x
    };
    const attrs = {
        axis
    };
    return (0, _engine.ENGINE).runKernel((0, _kernelNames.ArgMax), inputs, attrs);
}
Returns the indices of the maximum values along an `axis`.

The result has the same shape as `input` with the dimension along `axis`
removed.

```js
const x = tf.tensor1d([1, 2, 3]);

x.argMax().print();  // or tf.argMax(x)
```

```js
const x = tf.tensor2d([1, 2, 4, 3], [2, 2]);

const axis = 1;
x.argMax(axis).print();  // or tf.argMax(x, axis)
```

@param x The input tensor.
@param axis The dimension to reduce. Defaults to 0 (outer-most dimension).

@doc {heading: 'Operations', subheading: 'Reduction'}