• W
    Refine code and comments · ea7359c6
    wanghaoshuang 提交于
    1. Remove checking for num_neg_samples.
    2. Fix dims of Output(Cost) and Input(Bias).
    3. Renamed num_sampled_classes to num_neg_samples.
    4. Add TODO for add more distribution sampler.
    5. Init grad_data of bias by zero.
    6. Refine comments.
    7. Register a kernel for type double.
    ea7359c6
nce_op.h 8.2 KB