• Q
    check INFINITY in cross_entropy (#3287) · 4bbd05fd
    Qiao Longfei 提交于
    * check INFINITY in cross_entropy
    
    * fix error
    
    * use onehot_cross_entropy without GPU kernel
    
    * add support_gpu
    
    * fix allclose
    
    * fix name error and symplify code
    4bbd05fd
cross_entropy_op.cu 685 字节