* fix bug of ce loss when reduction != mean

* change function order

* modify comment

* minor fix