• Z
    [Cherry-Pick]Move valid check from python to kernel (#46980) · 8bfd45ad
    Zhang Zheng 提交于
    为了提升性能,将label的边界检查从python端转移到kernel内,减少额外op的调用,如min、max和同步拷贝等
        当前的模板参数IgnoreIndex仅在ignore_index取值范围在[0, dim)时才生效,但是当某个label值超出了边界,ignore_index等于该label,这种情况下是应该仍然能正常计算。虽然当前的计算逻辑在结果上不会出错,但逻辑上仍是有问题的,且模板参数IgnoreIndex是没有必要的
    8bfd45ad
loss.py 155.4 KB