asitstands commented on issue #10970: [MXNET-424] dtype option for multinomial URL: https://github.com/apache/incubator-mxnet/pull/10970#issuecomment-393026431 I added the check at `SampleMultinomialOpShape`. It ensures that the size of the last dimension of the input array is less than or equal to `2 << (the number of mantissa bits of dtype - 1)` for floating point types or `std::numeric_limits::max()` for integer types. A test for this is also added.
---------------------------------------------------------------- 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: [email protected] With regards, Apache Git Services
