title: Multi-dimensional np.argmax?
body: <p>I've got a 3D array of shape (n, n, g) and I'd need for every (n, n) the argmax, i.e. the result should be two index vectors (x, y) of length g each.</p>
  
  <p>The intuitive solution would be:</p>
  
  <pre class="lang-py prettyprint-override"><code>array = np.random.uniform(size=[5, 5, 1000])
  np.argmax(array, axis=[0, 1])
  </code></pre>
  
  <p>However, numpy does not support multiple axes as argument.</p>
  
  <p>Is there a solution to get this result anyways?</p>