The instructions for step 2.D say to use argmax “along the last axis”.
That means you need to include “axis = -1” in the arguments for argmax().
1 Like
The instructions for step 2.D say to use argmax “along the last axis”.
That means you need to include “axis = -1” in the arguments for argmax().