1. 29 9月, 2020 1 次提交
    • 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