• P
    add multinomial op (#27219) · 7cd2c13f
    pangyoki 提交于
    * add multinomial cpu kernel
    
    * fix C++ notype error
    
    * fix windows ci array len error
    
    * let array len be const
    
    * change array to vector
    
    * add cuda kernrl with num_distribution is 1, and not support replacement=False
    
    * add multinomial python api
    
    * support num_distribution different multinomial distributions
    
    * add multinomial python api unittest
    
    * change output dtype to int64
    
    * fix coverage prob
    
    * optimize format
    
    * fix dtype of output error, should be int64_t
    7cd2c13f
multinomial_op.cu 9.1 KB