The issue is raised in these lines of code since float32 is used in the multinomial operation and needs consecutive integer values, which are defined for <2**24 for float32. It seems float32 is strictly defined and casting inputs to float64 doesn’t solve the error. Could you create a feature request for your use case on GitHub and explain your use case a bit more, please, as this limitation might be relaxed.
1 Like