np_argmax
Find the indices of the maximum values in an array along a specified axis, or from the flattened array when no axis is given.
Instructions
Return the indices of the maximum values along an axis.
Input Schema
| Name | Required | Description | Default |
|---|---|---|---|
| array | Yes | The input array. | |
| axis | No | Axis along which to find argmax (default: None, flattened). |
Output Schema
| Name | Required | Description | Default |
|---|---|---|---|
| result | Yes |