Is there a 'argmax' method for NMatrix?
weili-git opened this issue · 1 comments
weili-git commented
By searching 'https://rdoc.info/github/SciRuby/nmatrix/NMatrix', I could only find 'max' method which returns the max value of certain dimension, but I need to get the index instead.
Thank you so much if someone could help!