Reduce Sum Pytorch