• Y
    Add New OP: gumbel_softmax (#35506) · 18eda6c3
    YuanRisheng 提交于
    * Add New Op: gumbel_softmax
    
    * Add New Op: gumbel_softmax
    
    * Add New Op: gumbel_softmax (amend)
    
    * add __main__ function in unit test
    
    * fix bugs when test in windows ci
    
    * update en docs
    
    * delete reletive error in unit test
    
    * delete relative error in unit test
    
    * set hard=True in unit test
    18eda6c3
gumbel_softmax_op.cc 4.5 KB