[GitHub] adrianloy opened a new issue #14226: Feature request: Argmax function should return an array if multiple entries with max value exist

[GitBox <gi...@apache.org>]

adrianloy opened a new issue #14226: Feature request: Argmax function should return an array if multiple entries with max value exist
URL: https://github.com/apache/incubator-mxnet/issues/14226
 
 
   Correct me if I am wrong, but currently it is quite a hassle to find a way in a network that should work in hybrid mode (so no asnumpy()) to get ALL indices where entries with maximum values exist. 
   However, there are quite a lot of functions that use list of indices, e.g. take(). But there are not many functions that will return list of indices, so functions like take() are often harder to use than it should be.
   
   So, feature request is to make argmax return a list of indices, like the argmax function in numpy or pytorch. This shouldn't slow down argmax as it doesnt change the time complexity.
   
   #6705 is basically the same problem. Maybe it would be possible to implement this request with #11061
    

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services