handle the case in acc_ops.sum when dim == 0, differentiating it from the case when dim is None (#64869)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/64869
handle the case in acc_ops.sum when dim == 0, differentiating it from the case when dim is None
Reviewed By: 842974287
Differential Revision: D30872739
fbshipit-source-id: 2755d3230804a16ef1c9289f804138c6dd7766b3