ch_ops.py (sum::compute_dr_wrt): bugfix for when axis is None and input is multidimensional.