[GitHub] [incubator-mxnet] Zha0q1 commented on issue #19753: TopK returns float32 indices

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

Zha0q1 commented on issue #19753:
URL: https://github.com/apache/incubator-mxnet/issues/19753#issuecomment-760557213


   > I think we should set the B[1] to int32/int64.
   
   I just realized there is a parameter `dtype` in topk to control the type of the indices output
   
   Also had a quick discussion with Rohit. We might force indices output to use int32/64 in 2.0. In v.1x we might want to keep the current behavior so that we do not break things


----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to 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



---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscribe@mxnet.apache.org
For additional commands, e-mail: issues-help@mxnet.apache.org