How would I set sum_axis with dot?