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: [email protected] --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
