mxnet-issues mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From GitBox <>
Subject [GitHub] [incubator-mxnet] Zha0q1 commented on issue #19753: TopK returns float32 indices
Date Fri, 15 Jan 2021 00:11:44 GMT

Zha0q1 commented on issue #19753:

   > 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
   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:

To unsubscribe, e-mail:
For additional commands, e-mail:

View raw message