title: single neuron layer after softmax (keras)
body: <p>I need to create a neural network (with keras) that has as last layer a single neuron that contains the index of the neuron with the maximum value prediction in the precedent softmax layer. </p>
  
  <p>For example my softmax layer gives as result this:</p>
  
  <pre><code>[0.1, 0.1, 0.7, 0.0, 0.05, 0.05]
  </code></pre>
  
  <p>And I want that the single neuron layer (after the softmax layer) gives as result <code>2</code> (considering a 0 based valutation).</p>
  
  <p>How can I do that?</p>