ZeroDivisionError in softmax cross entropy when I pass masked array
ZeroDivisionError in softmax cross entropy when I pass masked array: a task in MiMo-V2.6-RL-harbor-code: MiMo-V2.6-RL Code (Harbor) (Harbor dataset). When I mask two arrays and calculate softmax cross entroy , ZeroDivisionError occurs when masked array's data is I thought it should be checked in…
The task
When I mask two arrays and calculate `softmax_cross_entroy`, `ZeroDivisionError` occurs when masked array's data is `[]` I thought it should be checked in ndim and shape check at the beginning of the function, but the array's shape was something like `(0, 2, 2)`
Part of FineEnvs/MiMo-V2.6-RL-harbor-code.