How to write a cuda c argmax kernel like python does and let it decode one-hot matrix?

Question

Like the title suggested.

Here is the code.

Where show_data() and get_data() are functions to display and inport data.

I\'m not sure how to list through