Results 1 comments of McHz1s

Also occurred to me. Make a little change that: def func_name(...) pred_shape = y_pred.shape ... other codes ... dy_mean = 0 if pred_shape[2] == 1 or len(pred_shape) == 4 else...